add custom environment variable ASCEND_CUSTOM_PATH, for customized Ascend software installation

5 years ago · 0db74f80bb
parent 5369646b48
commit 0db74f80bb
1 changed files with 5 additions and 1 deletions
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@ -75,7 +75,11 @@ elseif(DEFINED ENV{D_LINK_PATH})
    find_library(resource libresource.so ${GE_LIB_PATH})
 else()
    # Ascend mode
-    set(HIAI_INSTALLED_DIR /usr/local/Ascend)
+    if(DEFINED ENV{ASCEND_CUSTOM_PATH})
+        set(HIAI_INSTALLED_DIR $ENV{D_LINK_PATH})
+    else()
+        set(HIAI_INSTALLED_DIR /usr/local/Ascend)
+    endif()
    set(HIAI_DRIVER_DIR ${HIAI_INSTALLED_DIR}/driver/lib64/common)
    set(HIAI_RUNTIME_DIR ${HIAI_INSTALLED_DIR}/fwkacllib/lib64)
    find_library(c_sec libc_sec.so ${HIAI_DRIVER_DIR})