Browse Source

!9 provide custom directory support for cases whereby Ascend software suite is installed out of the default location

Merge pull request !9 from yanghaoran/envmodify
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 4 years ago
parent
commit
ace35b9c93
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