From 8cdc7874968ebd157abc720e81f45a153c18ea89 Mon Sep 17 00:00:00 2001 From: wuweikang Date: Mon, 26 Oct 2020 22:27:59 +0800 Subject: [PATCH] libpath modification for atlas env --- CMakeLists.txt | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index 1288ea2c..8f8a39d6 100755 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -88,13 +88,21 @@ else() find_library(mmpa libmmpa.so ${ASCEND_DRIVER_DIR}) find_library(msprof libmsprof.so ${ASCEND_DRIVER_DIR}) find_library(msprofiler libmsprofiler.a ${ASCEND_RUNTIME_DIR}) - find_library(hccl libhccl.so ${ASCEND_RUNTIME_DIR}) find_library(runtime libruntime.so ${ASCEND_RUNTIME_DIR}) find_library(register libregister.so ${ASCEND_RUNTIME_DIR}) find_library(resource libresource.so ${ASCEND_RUNTIME_DIR}) find_library(error_manager liberror_manager.so ${ASCEND_RUNTIME_DIR}) find_library(adump_server libadump_server.a ${ASCEND_RUNTIME_DIR}) + # for Atlas env + set(ASCEND_TOOKIT_RUNTIME_DIR ${ASCEND_PATH}/ascend-toolkit/latest/fwkacllib/lib64) + find_library(msprofiler libmsprofiler.a ${ASCEND_TOOKIT_RUNTIME_DIR}) + find_library(hccl libhccl.so ${ASCEND_TOOKIT_RUNTIME_DIR}) + find_library(runtime libruntime.so ${ASCEND_TOOKIT_RUNTIME_DIR}) + find_library(register libregister.so ${ASCEND_TOOKIT_RUNTIME_DIR}) + find_library(resource libresource.so ${ASCEND_TOOKIT_RUNTIME_DIR}) + find_library(error_manager liberror_manager.so ${ASCEND_TOOKIT_RUNTIME_DIR}) + find_library(adump_server libadump_server.a ${ASCEND_TOOKIT_RUNTIME_DIR}) endif() # add compile flags