|
|
|
@ -219,7 +219,6 @@ if (ENABLE_D)
|
|
|
|
|
set(ASCEND_DRIVER_PATH ${ASCEND_PATH}/driver/lib64/common)
|
|
|
|
|
set(ASCEND_DRIVER_BACK_PATH ${ASCEND_PATH}/driver/lib64/driver)
|
|
|
|
|
set(ASCEND_RUNTIME_PATH ${ASCEND_PATH}/fwkacllib/lib64)
|
|
|
|
|
set(ASCEND_OPP_PATH ${ASCEND_PATH}/opp/op_impl/built-in/ai_core/tbe/op_tiling)
|
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
MESSAGE("USE DAV LIB PATH: ${ASCEND_PATH}")
|
|
|
|
@ -230,11 +229,11 @@ if (ENABLE_D)
|
|
|
|
|
find_library(DATATRANSFER datatransfer HINTS ${ASCEND_RUNTIME_PATH} ${ASCEND_DRIVER_BACK_PATH})
|
|
|
|
|
find_library(PROFILING msprofiler ${ASCEND_RUNTIME_PATH})
|
|
|
|
|
find_library(PROFILING_SHARED msprof ${ASCEND_DRIVER_PATH})
|
|
|
|
|
find_library(OPTILING optiling ${ASCEND_OPP_PATH})
|
|
|
|
|
find_library(REGISTER register ${ASCEND_RUNTIME_PATH})
|
|
|
|
|
add_library(ms_profile SHARED ${PROFILING})
|
|
|
|
|
set_target_properties(ms_profile PROPERTIES LINKER_LANGUAGE CXX)
|
|
|
|
|
target_link_libraries(ms_profile -Wl,--start-group ${PROFILING_SHARED} ${PROFILING} mindspore::protobuf -Wl,--end-group)
|
|
|
|
|
target_link_libraries(mindspore ms_profile ge_runtime ${CCE_LIB} ${RUNTIME_LIB} ${TSDCLIENT} ${HCCL} ${DATATRANSFER} ${OPTILING})
|
|
|
|
|
target_link_libraries(mindspore ms_profile ge_runtime ${CCE_LIB} ${RUNTIME_LIB} ${TSDCLIENT} ${HCCL} ${DATATRANSFER} ${REGISTER})
|
|
|
|
|
target_link_libraries(mindspore -Wl,--start-group proto_input mindspore::protobuf -Wl,--end-group)
|
|
|
|
|
elseif (CMAKE_SYSTEM_NAME MATCHES "Windows")
|
|
|
|
|
target_link_libraries(mindspore -Wl,--start-group proto_input mindspore::protobuf mindspore::sentencepiece -Wl,--end-group)
|
|
|
|
@ -265,7 +264,6 @@ if (ENABLE_D)
|
|
|
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64)
|
|
|
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/fwkacllib/lib64)
|
|
|
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/add-ons)
|
|
|
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/op_tiling)
|
|
|
|
|
elseif (ENABLE_GPU)
|
|
|
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/cuda/lib64)
|
|
|
|
|
endif ()
|
|
|
|
@ -324,7 +322,6 @@ add_library(inference SHARED
|
|
|
|
|
${LOAD_ONNX_SRC}
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
set_target_properties(inference PROPERTIES INSTALL_RPATH ${MINDSPORE_RPATH})
|
|
|
|
|
target_link_libraries(inference PRIVATE ${PYTHON_LIBRARIES} ${SECUREC_LIBRARY}
|
|
|
|
|
-Wl,--whole-archive mindspore proto_input -Wl,--no-whole-archive mindspore_gvar)
|
|
|
|
|
|
|
|
|
|