Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
正统之独孤求败
mindspore
提交
5583532e
M
mindspore
项目概览
正统之独孤求败
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
5583532e
编写于
4月 02, 2020
作者:
J
jjfeing
提交者:
高东海
4月 08, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
tr5 run package update
上级
b9ccd6d5
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
27 addition
and
22 deletion
+27
-22
cmake/dependency_graphengine.cmake
cmake/dependency_graphengine.cmake
+13
-12
mindspore/_extends/parallel_compile/tbe_compiler/common.py
mindspore/_extends/parallel_compile/tbe_compiler/common.py
+1
-1
mindspore/ccsrc/CMakeLists.txt
mindspore/ccsrc/CMakeLists.txt
+13
-9
未找到文件。
cmake/dependency_graphengine.cmake
浏览文件 @
5583532e
...
...
@@ -38,19 +38,20 @@ elseif (DEFINED ENV{D_LINK_PATH})
find_library
(
cce libcce.so
${
GE_LIB_PATH
}
)
find_library
(
resource libresource.so
${
GE_LIB_PATH
}
)
else
()
set
(
HIAI_INSTALLED_DIR /usr/local/HiAI
)
set
(
HIAI_DRIVER_DIR
${
HIAI_INSTALLED_DIR
}
/driver/lib64
)
set
(
HIAI_RUNTIME_DIR
${
HIAI_INSTALLED_DIR
}
/runtime/lib64
)
find_library
(
c_sec libc_sec.so
${
HIAI_DRIVER_DIR
}
)
find_library
(
slog libslog.so
${
HIAI_DRIVER_DIR
}
)
find_library
(
mmpa libmmpa.so
${
HIAI_DRIVER_DIR
}
)
# Ascend mode
set
(
ASCEND_PATH /usr/local/Ascend
)
set
(
ASCEND_DRIVER_PATH
${
ASCEND_PATH
}
/driver/lib64/common
)
set
(
ASCEND_RUNTIME_PATH
${
ASCEND_PATH
}
/fwkacllib/lib64
)
find_library
(
cce libcce.so
${
HIAI_RUNTIME_DIR
}
)
find_library
(
hccl libhccl.so
${
HIAI_RUNTIME_DIR
}
)
find_library
(
runtime libruntime.so
${
HIAI_RUNTIME_DIR
}
)
find_library
(
msprof libmsprof.so
${
HIAI_RUNTIME_DIR
}
)
find_library
(
register libregister.so
${
HIAI_RUNTIME_DIR
}
)
find_library
(
resource libresource.so
${
HIAI_RUNTIME_DIR
}
)
find_library
(
c_sec libc_sec.so
${
ASCEND_DRIVER_PATH
}
)
find_library
(
slog libslog.so
${
ASCEND_DRIVER_PATH
}
)
find_library
(
mmpa libmmpa.so
${
ASCEND_DRIVER_PATH
}
)
find_library
(
cce libcce.so
${
ASCEND_RUNTIME_PATH
}
)
find_library
(
hccl libhccl.so
${
ASCEND_RUNTIME_PATH
}
)
find_library
(
runtime libruntime.so
${
ASCEND_RUNTIME_PATH
}
)
find_library
(
msprof libmsprof.so
${
ASCEND_RUNTIME_PATH
}
)
find_library
(
register libregister.so
${
ASCEND_RUNTIME_PATH
}
)
find_library
(
resource libresource.so
${
ASCEND_RUNTIME_PATH
}
)
endif
()
# compile libraries from following directories
...
...
mindspore/_extends/parallel_compile/tbe_compiler/common.py
浏览文件 @
5583532e
...
...
@@ -50,7 +50,7 @@ def get_build_in_impl_path():
tbe_impl_path
=
os
.
environ
.
get
(
"TBE_IMPL_PATH"
)
if
tbe_impl_path
is
None
:
default_install_path
=
'/usr/local/HiAI/runtime/ops/op_impl/built-in/ai_core/tbe/'
backup_install_path
=
'/usr/local/Ascend/
Ascend/
opp/op_impl/built-in/ai_core/tbe/'
backup_install_path
=
'/usr/local/Ascend/opp/op_impl/built-in/ai_core/tbe/'
if
os
.
path
.
exists
(
default_install_path
):
tbe_impl_path
=
default_install_path
elif
os
.
path
.
exists
(
backup_install_path
):
...
...
mindspore/ccsrc/CMakeLists.txt
浏览文件 @
5583532e
...
...
@@ -279,15 +279,17 @@ if(ENABLE_D)
endif
()
else
()
MESSAGE
(
"use system default lib"
)
set
(
D_LIB_PATH
"/usr/local/HiAI/runtime/lib64/"
)
set
(
ASCEND_PATH /usr/local/Ascend
)
set
(
ASCEND_DRIVER_PATH
${
ASCEND_PATH
}
/driver/lib64/common
)
set
(
ASCEND_RUNTIME_PATH
${
ASCEND_PATH
}
/fwkacllib/lib64
)
endif
()
MESSAGE
(
"USE DAV LIB PATH:
${
D_LIB
_PATH
}
"
)
find_library
(
HCCL hccl
${
D_LIB
_PATH
}
)
find_library
(
CCE_LIB cce
${
D_LIB
_PATH
}
)
find_library
(
RUNTIME_LIB runtime
${
D_LIB
_PATH
}
)
find_library
(
TSDCLIENT tsdclient
${
D_LIB
_PATH
}
)
find_library
(
PROFILING msprof
${
D_LIB
_PATH
}
)
MESSAGE
(
"USE DAV LIB PATH:
${
ASCEND
_PATH
}
"
)
find_library
(
HCCL hccl
${
ASCEND_RUNTIME
_PATH
}
)
find_library
(
CCE_LIB cce
${
ASCEND_RUNTIME
_PATH
}
)
find_library
(
RUNTIME_LIB runtime
${
ASCEND_RUNTIME
_PATH
}
)
find_library
(
TSDCLIENT tsdclient
${
ASCEND_RUNTIME
_PATH
}
)
find_library
(
PROFILING msprof
${
ASCEND_DRIVER
_PATH
}
)
target_link_libraries
(
mindspore ge_runtime
${
CCE_LIB
}
${
RUNTIME_LIB
}
${
TSDCLIENT
}
${
PROFILING
}
${
HCCL
}
${
TSDCLIENT
}
)
endif
()
...
...
@@ -481,12 +483,14 @@ add_dependencies(add_ms_lib _c_expression)
if
(
NOT ENABLE_GE
)
if
(
ENABLE_D
)
set
(
ASCEND_PATH /usr/local/Ascend
)
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
}
COMMAND cp
${
MS_CCSRC_BUILD_PATH
}
/../../graphengine/src/ge/common/libge_common.so
${
MS_LIB_PATH
}
COMMAND cp
${
MS_CCSRC_BUILD_PATH
}
/../../graphengine/src/ge/ge_runtime/libge_runtime.so
${
MS_LIB_PATH
}
COMMAND cp
/usr/local/HiAI/driver/lib64
/libslog.so
${
MS_LIB_PATH
}
COMMAND cp
/usr/local/HiAI/driver/lib64
/libc_sec.so
${
MS_LIB_PATH
}
COMMAND cp
${
ASCEND_DRIVER_PATH
}
/libslog.so
${
MS_LIB_PATH
}
COMMAND cp
${
ASCEND_DRIVER_PATH
}
/libc_sec.so
${
MS_LIB_PATH
}
)
add_dependencies
(
add_ge_lib add_ms_lib
)
add_dependencies
(
add_ge_lib graph
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录