|
|
|
@@ -77,13 +77,14 @@ set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -Wno-unused-private-field") |
|
|
|
if (SUPPORT_GPU) |
|
|
|
add_subdirectory(runtime/kernel/opencl) |
|
|
|
target_link_libraries(mindspore-lite cpu_kernel_mid opencl_kernel_mid nnacl) |
|
|
|
target_link_libraries(mindspore-lite_static cpu_kernel_mid opencl_kernel_mid nnacl) |
|
|
|
target_link_libraries(mindspore-lite_static cpu_kernel_mid opencl_kernel_mid nnacl_mid) |
|
|
|
else () |
|
|
|
target_link_libraries(mindspore-lite cpu_kernel_mid nnacl) |
|
|
|
target_link_libraries(mindspore-lite_static cpu_kernel_mid nnacl) |
|
|
|
target_link_libraries(mindspore-lite_static cpu_kernel_mid nnacl_mid) |
|
|
|
endif () |
|
|
|
if (PLATFORM_ARM32 OR PLATFORM_ARM64) |
|
|
|
target_link_libraries(mindspore-lite log) |
|
|
|
target_link_libraries(mindspore-lite_static log) |
|
|
|
endif () |
|
|
|
if (BUILD_MINDDATA STREQUAL "lite") |
|
|
|
target_link_libraries(mindspore-lite minddata_eager_mid minddata-lite) |
|
|
|
|