|
|
@@ -29,6 +29,7 @@ set(ASCEND_RUNTIME_DIR ${ASCEND_DIR}/fwkacllib/lib64) |
|
|
|
set(ASCEND_ATC_DIR ${ASCEND_DIR}/atc/lib64) |
|
|
|
set(ASCEND_ACL_DIR ${ASCEND_DIR}/acllib/lib64) |
|
|
|
set(STATIC_ACL_LIB ${ASCEND_ACL_DIR}) |
|
|
|
set(ASCEND_RUNTIME_PATH ${ASCEND_RUNTIME_DIR} ${ASCEND_ACL_DIR} ${ASCEND_ATC_DIR}) |
|
|
|
|
|
|
|
set(ASCEND_MS_RUNTIME_PATH ${ASCEND_RUNTIME_DIR} ${ASCEND_ACL_DIR} ${ASCEND_ATC_DIR}) |
|
|
|
set(ASCEND_MS_DRIVER_PATH ${ASCEND_DRIVER_DIR} ${ASCEND_DRIVER_COMMON_DIR}) |
|
|
@@ -88,7 +89,6 @@ else () |
|
|
|
find_module(hccl libhccl.so ${GE_LIB_PATH}) |
|
|
|
find_module(adump_server libadump_server.a ${GE_LIB_PATH}) |
|
|
|
find_module(runtime libruntime.so ${GE_LIB_PATH}) |
|
|
|
find_module(runtime_compile libruntime_compile.so ${GE_LIB_PATH}) |
|
|
|
find_module(resource libresource.so ${GE_LIB_PATH}) |
|
|
|
find_module(ascend_hal_stub libascend_hal.so ${GE_LIB_PATH}) |
|
|
|
find_module(msprofiler_fwk_ext libmsprofiler_fwk.a ${GE_LIB_PATH}) |
|
|
@@ -108,7 +108,6 @@ else () |
|
|
|
elseif(PLATFORM STREQUAL "inference") |
|
|
|
find_module(adump_server libadump_server.a ${ASCEND_ACL_DIR}) |
|
|
|
find_module(runtime libruntime.so ${ASCEND_ACL_DIR}) |
|
|
|
find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR}) |
|
|
|
find_module(msprofiler_ext libmsprofiler.a ${ASCEND_ACL_DIR}) |
|
|
|
if(PRODUCT STREQUAL "flr3") |
|
|
|
elseif(PRODUCT STREQUAL "flr1") |
|
|
@@ -120,10 +119,9 @@ else () |
|
|
|
endif() |
|
|
|
elseif(PLATFORM STREQUAL "all") |
|
|
|
find_module(adump_server libadump_server.a ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(runtime libruntime.so ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(runtime libruntime.so ${ASCEND_RUNTIME_PATH}) |
|
|
|
find_module(msprofiler_fwk_ext libmsprofiler_fwk.a ${ASCEND_RUNTIME_DIR}) |
|
|
|
find_module(ascend_hal_stub libascend_hal.so ${ASCEND_DRIVER_DIR}) |
|
|
|
find_module(runtime_compile libruntime_compile.so ${ASCEND_ATC_DIR}) |
|
|
|
find_module(msprofiler_ext libmsprofiler.a ${ASCEND_ACL_DIR}) |
|
|
|
else() |
|
|
|
message(STATUS "PLATFORM param is invalid, should be train or inference, you choose nothing!") |
|
|
|