Browse Source

add custom environment variable ASCEND_CUSTOM_PATH, for customized Ascend software installation

tags/v0.2.0-alpha
yanghaoran 4 years ago
parent
commit
0db74f80bb
1 changed files with 5 additions and 1 deletions
  1. +5
    -1
      CMakeLists.txt

+ 5
- 1
CMakeLists.txt View File

@@ -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})


Loading…
Cancel
Save