|
|
@ -162,6 +162,7 @@ set(SUB_COMP
|
|
|
|
backend/kernel_compiler
|
|
|
|
backend/kernel_compiler
|
|
|
|
backend/session
|
|
|
|
backend/session
|
|
|
|
runtime/device
|
|
|
|
runtime/device
|
|
|
|
|
|
|
|
runtime/hccl_adapter
|
|
|
|
frontend/optimizer
|
|
|
|
frontend/optimizer
|
|
|
|
frontend/parallel
|
|
|
|
frontend/parallel
|
|
|
|
frontend/operator
|
|
|
|
frontend/operator
|
|
|
@ -239,17 +240,16 @@ if (ENABLE_D)
|
|
|
|
find_library(PROFILING_SHARED msprof ${ASCEND_DRIVER_PATH})
|
|
|
|
find_library(PROFILING_SHARED msprof ${ASCEND_DRIVER_PATH})
|
|
|
|
find_library(REGISTER register ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
|
|
find_library(REGISTER register ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
|
|
find_library(OPTILING optiling ${ASCEND_OPP_PATH})
|
|
|
|
find_library(OPTILING optiling ${ASCEND_OPP_PATH})
|
|
|
|
|
|
|
|
# hccl_adpter
|
|
|
|
|
|
|
|
find_library(HCCL_ADPTER hcom_graph_adaptor ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
|
|
|
|
|
|
find_library(HCCL_BUILDER hcom_opskernel_builder ${ASCEND_RUNTIME_PATH}/plugin/opskernel ${ASCEND_TOOLKIT_RUNTIME_PATH}/plugin/opskernel)
|
|
|
|
|
|
|
|
|
|
|
|
add_library(ms_profile SHARED ${PROFILING})
|
|
|
|
add_library(ms_profile SHARED ${PROFILING})
|
|
|
|
set_target_properties(ms_profile PROPERTIES LINKER_LANGUAGE CXX)
|
|
|
|
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(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}
|
|
|
|
target_link_libraries(mindspore ms_profile ge_runtime ${CCE_LIB} ${RUNTIME_LIB} ${TSDCLIENT} ${HCCL} ${DATATRANSFER}
|
|
|
|
${REGISTER} -Wl,--no-as-needed ${OPTILING})
|
|
|
|
${HCCL_ADPTER} ${REGISTER} -Wl,--no-as-needed ${OPTILING} ${HCCL_BUILDER})
|
|
|
|
target_link_libraries(mindspore -Wl,--start-group proto_input mindspore::protobuf -Wl,--end-group)
|
|
|
|
target_link_libraries(mindspore -Wl,--start-group proto_input mindspore::protobuf -Wl,--end-group)
|
|
|
|
# hccl_adpter
|
|
|
|
|
|
|
|
find_library(HCCL_ADPTER hcom_graph_adaptor ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
|
|
|
|
|
|
find_library(HCCL_BUILDER hcom_opskernel_builder ${ASCEND_RUNTIME_PATH}/plugin/opskernel ${ASCEND_TOOLKIT_RUNTIME_PATH}/plugin/opskernel)
|
|
|
|
|
|
|
|
add_subdirectory(runtime/hccl_adapter)
|
|
|
|
|
|
|
|
target_link_libraries(hccl_adapter PRIVATE mindspore ${HCCL_ADPTER} ${REGISTER} -Wl,--no-as-needed ${HCCL_BUILDER})
|
|
|
|
|
|
|
|
elseif (CMAKE_SYSTEM_NAME MATCHES "Windows")
|
|
|
|
elseif (CMAKE_SYSTEM_NAME MATCHES "Windows")
|
|
|
|
target_link_libraries(mindspore -Wl,--start-group proto_input mindspore::protobuf mindspore::sentencepiece -Wl,--end-group)
|
|
|
|
target_link_libraries(mindspore -Wl,--start-group proto_input mindspore::protobuf mindspore::sentencepiece -Wl,--end-group)
|
|
|
|
else ()
|
|
|
|
else ()
|
|
|
@ -288,16 +288,10 @@ if (ENABLE_D)
|
|
|
|
elseif (ENABLE_GPU)
|
|
|
|
elseif (ENABLE_GPU)
|
|
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/cuda/lib64)
|
|
|
|
set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/cuda/lib64)
|
|
|
|
endif ()
|
|
|
|
endif ()
|
|
|
|
set(HCCL_ADPT_RPATH ${ORIGIN_PATH}:${MINDSPORE_RPATH})
|
|
|
|
|
|
|
|
set(MINDSPORE_RPATH ${ORIGIN_PATH}/lib:${MINDSPORE_RPATH})
|
|
|
|
set(MINDSPORE_RPATH ${ORIGIN_PATH}/lib:${MINDSPORE_RPATH})
|
|
|
|
|
|
|
|
|
|
|
|
set_target_properties(_c_expression PROPERTIES INSTALL_RPATH ${MINDSPORE_RPATH})
|
|
|
|
set_target_properties(_c_expression PROPERTIES INSTALL_RPATH ${MINDSPORE_RPATH})
|
|
|
|
|
|
|
|
|
|
|
|
if (ENABLE_D)
|
|
|
|
|
|
|
|
set_target_properties(hccl_adapter PROPERTIES INSTALL_RPATH ${HCCL_ADPT_RPATH})
|
|
|
|
|
|
|
|
target_link_libraries(_c_expression PRIVATE hccl_adapter)
|
|
|
|
|
|
|
|
endif ()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if (CMAKE_SYSTEM_NAME MATCHES "Windows")
|
|
|
|
if (CMAKE_SYSTEM_NAME MATCHES "Windows")
|
|
|
|
target_link_libraries(mindspore mindspore::pybind11_module)
|
|
|
|
target_link_libraries(mindspore mindspore::pybind11_module)
|
|
|
|
target_link_libraries(mindspore mindspore_gvar)
|
|
|
|
target_link_libraries(mindspore mindspore_gvar)
|
|
|
@ -362,7 +356,6 @@ if (ENABLE_D)
|
|
|
|
find_library(adump_server libadump_server.a ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
|
|
find_library(adump_server libadump_server.a ${ASCEND_RUNTIME_PATH} ${ASCEND_TOOLKIT_RUNTIME_PATH})
|
|
|
|
target_link_libraries(_c_expression PRIVATE ${adump_server})
|
|
|
|
target_link_libraries(_c_expression PRIVATE ${adump_server})
|
|
|
|
target_link_libraries(inference PRIVATE ${adump_server})
|
|
|
|
target_link_libraries(inference PRIVATE ${adump_server})
|
|
|
|
target_link_libraries(inference PRIVATE mindspore_core hccl_adapter)
|
|
|
|
|
|
|
|
endif()
|
|
|
|
endif()
|
|
|
|
|
|
|
|
|
|
|
|
if (ENABLE_CPU)
|
|
|
|
if (ENABLE_CPU)
|
|
|
|