From ac798d87dc90d1722fe73bb4362d2d897421dd4b Mon Sep 17 00:00:00 2001 From: yanghaoran Date: Wed, 23 Sep 2020 11:07:01 +0800 Subject: [PATCH] modify atlas installer package default directory, due to changes of installer packages --- mindspore/_version_check.py | 6 +++--- mindspore/ccsrc/CMakeLists.txt | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/_version_check.py b/mindspore/_version_check.py index a8c7b573cd..6d07b4a221 100644 --- a/mindspore/_version_check.py +++ b/mindspore/_version_check.py @@ -122,12 +122,12 @@ class AscendEnvChecker(EnvChecker): def __init__(self): self.version = ["1.75.22.0.220"] - atlas_fwk_version = "/usr/local/Ascend/nnae/latest/fwkacllib/version.info" + atlas_fwk_version = "/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/version.info" hisi_fwk_version = "/usr/local/Ascend/fwkacllib/version.info" if os.path.exists(atlas_fwk_version): # atlas default path - self.fwk_path = "/usr/local/Ascend/nnae/latest/fwkacllib" - self.op_impl_path = "/usr/local/Ascend/nnae/latest/opp/op_impl/built-in/ai_core/tbe" + self.fwk_path = "/usr/local/Ascend/ascend-toolkit/latest/fwkacllib" + self.op_impl_path = "/usr/local/Ascend/ascend-toolkit/latest/opp/op_impl/built-in/ai_core/tbe" self.tbe_path = self.fwk_path + "/lib64" self.cce_path = self.fwk_path + "/ccec_compiler/bin" self.fwk_version = atlas_fwk_version diff --git a/mindspore/ccsrc/CMakeLists.txt b/mindspore/ccsrc/CMakeLists.txt index 3dd8dc9650..b53a780113 100644 --- a/mindspore/ccsrc/CMakeLists.txt +++ b/mindspore/ccsrc/CMakeLists.txt @@ -258,7 +258,7 @@ else () endif () set(MINDSPORE_RPATH ${ORIGIN_PATH}/lib) if (ENABLE_D) - set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/nnae/latest/fwkacllib/lib64) + set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/ascend-toolkit/latest/fwkacllib/lib64) set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/fwkacllib/lib64) set(MINDSPORE_RPATH ${MINDSPORE_RPATH}:/usr/local/Ascend/add-ons) elseif (ENABLE_GPU)