|
|
|
@ -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)
|
|
|
|
|