Browse Source

add custom environment variable ASCEND_CUSTOM_PATH for customized Ascend software installation

tags/v0.2.0-alpha
yanghaoran 6 years ago
parent
commit
576f65ad48
3 changed files with 16 additions and 4 deletions
  1. +5
    -1
      cmake/dependency_graphengine.cmake
  2. +1
    -1
      graphengine
  3. +10
    -2
      mindspore/ccsrc/CMakeLists.txt

+ 5
- 1
cmake/dependency_graphengine.cmake View File

@@ -39,7 +39,11 @@ elseif (DEFINED ENV{D_LINK_PATH})
find_library(resource libresource.so ${GE_LIB_PATH})
else()
# Ascend mode
set(ASCEND_PATH /usr/local/Ascend)
if(DEFINED ENV{ASCEND_CUSTOM_PATH})
set(ASCEND_PATH $ENV{ASCEND_CUSTOM_PATH})
else()
set(ASCEND_PATH /usr/local/Ascend)
endif()
set(ASCEND_DRIVER_PATH ${ASCEND_PATH}/driver/lib64/common)
set(ASCEND_RUNTIME_PATH ${ASCEND_PATH}/fwkacllib/lib64)
find_library(c_sec libc_sec.so ${ASCEND_DRIVER_PATH})


+ 1
- 1
graphengine

@@ -1 +1 @@
Subproject commit 5369646b489114b380a7b5208ddd6e632acb447f
Subproject commit 5bd0dc1ed59a9ec4ea6a602bf1385c59d845f922

+ 10
- 2
mindspore/ccsrc/CMakeLists.txt View File

@@ -296,7 +296,11 @@ if(ENABLE_D)
endif()
else()
MESSAGE("use system default lib")
set(ASCEND_PATH /usr/local/Ascend)
if(DEFINED ENV{ASCEND_CUSTOM_PATH})
set(ASCEND_PATH $ENV{ASCEND_CUSTOM_PATH})
else()
set(ASCEND_PATH /usr/local/Ascend)
endif()
set(ASCEND_DRIVER_PATH ${ASCEND_PATH}/driver/lib64/common)
set(ASCEND_RUNTIME_PATH ${ASCEND_PATH}/fwkacllib/lib64)
endif()
@@ -500,7 +504,11 @@ add_dependencies(add_ms_lib _c_expression)

if (NOT ENABLE_GE)
if (ENABLE_D)
set(ASCEND_PATH /usr/local/Ascend)
if(DEFINED ENV{ASCEND_CUSTOM_PATH})
set(ASCEND_PATH $ENV{ASCEND_CUSTOM_PATH})
else()
set(ASCEND_PATH /usr/local/Ascend)
endif()
set(ASCEND_DRIVER_PATH ${ASCEND_PATH}/driver/lib64/common)
add_custom_target(add_ge_lib ALL
COMMAND cp ${MS_CCSRC_BUILD_PATH}/../../graphengine/src/common/graph/libgraph.so ${MS_LIB_PATH}


Loading…
Cancel
Save