|
|
@@ -301,6 +301,7 @@ if(ENABLE_D) |
|
|
set(ASCEND_PATH /usr/local/Ascend) |
|
|
set(ASCEND_PATH /usr/local/Ascend) |
|
|
endif() |
|
|
endif() |
|
|
set(ASCEND_DRIVER_PATH ${ASCEND_PATH}/driver/lib64/common) |
|
|
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_RUNTIME_PATH ${ASCEND_PATH}/fwkacllib/lib64) |
|
|
endif() |
|
|
endif() |
|
|
|
|
|
|
|
|
@@ -308,7 +309,7 @@ if(ENABLE_D) |
|
|
find_library(HCCL hccl ${ASCEND_RUNTIME_PATH}) |
|
|
find_library(HCCL hccl ${ASCEND_RUNTIME_PATH}) |
|
|
find_library(CCE_LIB cce ${ASCEND_RUNTIME_PATH}) |
|
|
find_library(CCE_LIB cce ${ASCEND_RUNTIME_PATH}) |
|
|
find_library(RUNTIME_LIB runtime ${ASCEND_RUNTIME_PATH}) |
|
|
find_library(RUNTIME_LIB runtime ${ASCEND_RUNTIME_PATH}) |
|
|
find_library(TSDCLIENT tsdclient ${ASCEND_RUNTIME_PATH}) |
|
|
|
|
|
|
|
|
find_library(TSDCLIENT tsdclient HINTS ${ASCEND_RUNTIME_PATH} ${ASCEND_DRIVER_BACK_PATH}) |
|
|
find_library(PROFILING msprof ${ASCEND_DRIVER_PATH}) |
|
|
find_library(PROFILING msprof ${ASCEND_DRIVER_PATH}) |
|
|
target_link_libraries(mindspore ge_runtime ${CCE_LIB} ${RUNTIME_LIB} ${TSDCLIENT} ${PROFILING} ${HCCL} ${TSDCLIENT}) |
|
|
target_link_libraries(mindspore ge_runtime ${CCE_LIB} ${RUNTIME_LIB} ${TSDCLIENT} ${PROFILING} ${HCCL} ${TSDCLIENT}) |
|
|
endif() |
|
|
endif() |
|
|
|