Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
9ce5c972
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
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看板
提交
9ce5c972
编写于
7月 07, 2020
作者:
Y
yanghaoran
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
synchronize Ascend software suite 07 Jul 2020
上级
ba393c83
变更
7
隐藏空白更改
内联
并排
Showing
7 changed file
with
10 addition
and
14 deletion
+10
-14
cmake/dependency_graphengine.cmake
cmake/dependency_graphengine.cmake
+1
-0
cmake/package.cmake
cmake/package.cmake
+3
-0
graphengine
graphengine
+1
-1
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
+2
-1
mindspore/ccsrc/utils/context/ms_context.cc
mindspore/ccsrc/utils/context/ms_context.cc
+0
-10
tests/ut/cpp/stub/hccl/hccl_stub.cc
tests/ut/cpp/stub/hccl/hccl_stub.cc
+2
-1
未找到文件。
cmake/dependency_graphengine.cmake
浏览文件 @
9ce5c972
...
...
@@ -15,6 +15,7 @@ include(${GE_SOURCE_DIR}/cmake/external_libs/securec.cmake)
if
(
NOT ENABLE_D
)
set
(
GE_PREBUILD_PATH
${
GE_SOURCE_DIR
}
/third_party/prebuild/
${
CMAKE_HOST_SYSTEM_PROCESSOR
}
)
find_library
(
slog libslog.so
${
GE_PREBUILD_PATH
}
)
find_library
(
error_manager liberror_manager.so
${
GE_PREBUILD_PATH
}
)
elseif
(
DEFINED ENV{D_LINK_PATH}
)
set
(
GE_LIB_PATH $ENV{D_LINK_PATH}
)
set
(
GE_SYS_ARCH
""
)
...
...
cmake/package.cmake
浏览文件 @
9ce5c972
...
...
@@ -156,6 +156,7 @@ if (NOT ENABLE_GE)
set
(
ASCEND_PATH /usr/local/Ascend
)
endif
()
set
(
ASCEND_DRIVER_PATH
${
ASCEND_PATH
}
/driver/lib64/common
)
set
(
ASCEND_FWK_PATH
${
ASCEND_PATH
}
/fwkacllib/lib64
)
install
(
FILES
...
...
@@ -164,6 +165,7 @@ if (NOT ENABLE_GE)
${
CMAKE_BINARY_DIR
}
/graphengine/src/ge/ge_runtime/libge_runtime.so
${
ASCEND_DRIVER_PATH
}
/libslog.so
${
ASCEND_DRIVER_PATH
}
/libc_sec.so
${
ASCEND_FWK_PATH
}
/liberror_manager.so
DESTINATION
${
INSTALL_LIB_DIR
}
COMPONENT mindspore
)
...
...
@@ -172,6 +174,7 @@ if (NOT ENABLE_GE)
FILES
${
CMAKE_BINARY_DIR
}
/graphengine/src/common/graph/libgraph.so
${
CMAKE_SOURCE_DIR
}
/graphengine/third_party/prebuild/
${
CMAKE_HOST_SYSTEM_PROCESSOR
}
/libslog.so
${
CMAKE_SOURCE_DIR
}
/graphengine/third_party/prebuild/
${
CMAKE_HOST_SYSTEM_PROCESSOR
}
/liberror_manager.so
${
CMAKE_SOURCE_DIR
}
/build/graphengine/libc_sec.so
DESTINATION
${
INSTALL_LIB_DIR
}
COMPONENT mindspore
...
...
graphengine
@
4084909d
Subproject commit
8891f0546c4a250095ff68e1262f58772b938fd9
Subproject commit
4084909d62c159da6ba316f61ad3d02a4857b34b
mindspore/_extends/parallel_compile/tbe_compiler/common.py
浏览文件 @
9ce5c972
...
...
@@ -40,7 +40,7 @@ def get_ddk_version():
with
open
(
backup_ddk_info_file
,
"r"
)
as
fp
:
ddk_version
=
json
.
load
(
fp
)[
"VERSION"
]
else
:
ddk_version
=
"
1.60.T17.B83
0"
ddk_version
=
"
Ascend91
0"
return
ddk_version
...
...
mindspore/ccsrc/CMakeLists.txt
浏览文件 @
9ce5c972
...
...
@@ -192,8 +192,9 @@ if (ENABLE_D)
find_library
(
CCE_LIB cce
${
ASCEND_RUNTIME_PATH
}
)
find_library
(
RUNTIME_LIB runtime
${
ASCEND_RUNTIME_PATH
}
)
find_library
(
TSDCLIENT tsdclient HINTS
${
ASCEND_RUNTIME_PATH
}
${
ASCEND_DRIVER_BACK_PATH
}
)
find_library
(
DATATRANSFER datatransfer HINTS
${
ASCEND_RUNTIME_PATH
}
${
ASCEND_DRIVER_BACK_PATH
}
)
find_library
(
PROFILING msprof
${
ASCEND_DRIVER_PATH
}
)
target_link_libraries
(
mindspore ge_runtime
${
CCE_LIB
}
${
RUNTIME_LIB
}
${
TSDCLIENT
}
${
PROFILING
}
${
HCCL
}
${
TSDCLIENT
}
)
target_link_libraries
(
mindspore ge_runtime
${
CCE_LIB
}
${
RUNTIME_LIB
}
${
TSDCLIENT
}
${
PROFILING
}
${
HCCL
}
${
DATATRANSFER
}
)
endif
()
# link protobuf
...
...
mindspore/ccsrc/utils/context/ms_context.cc
浏览文件 @
9ce5c972
...
...
@@ -186,28 +186,18 @@ bool MsContext::OpenTsd() {
}
MS_LOG
(
INFO
)
<<
"Device id = "
<<
device_id
<<
", rank size = "
<<
rank_size
<<
"."
;
#if (defined(ENABLE_TDTQUE) && defined(ENABLE_GE))
int32_t
initStatus
=
tdt
::
TdtHostInit
(
device_id
);
if
(
initStatus
!=
TDT_OK_CODE
)
{
MS_LOG
(
EXCEPTION
)
<<
"Init tsd failed, status = "
<<
initStatus
<<
"."
;
return
false
;
}
tdt_print_
=
std
::
thread
(
TensorPrint
());
#endif
TDT_StatusT
status
=
tdt
::
TsdClient
::
GetInstance
()
->
Open
(
device_id
,
rank_size
);
if
(
status
!=
TDT_OK
)
{
MS_LOG
(
EXCEPTION
)
<<
"Device "
<<
device_id
<<
" is occupied, open tsd failed, status = "
<<
status
<<
"."
;
return
false
;
}
tsd_ref_
++
;
#if (defined(ENABLE_TDTQUE) && !defined(ENABLE_GE))
int32_t
initStatus
=
tdt
::
TdtHostInit
(
device_id
);
if
(
initStatus
!=
TDT_OK_CODE
)
{
MS_LOG
(
EXCEPTION
)
<<
"Init tsd failed, status = "
<<
initStatus
<<
"."
;
return
false
;
}
tdt_print_
=
std
::
thread
(
TensorPrint
());
#endif
MS_LOG
(
INFO
)
<<
"Open and init tsd successful, tsd reference = "
<<
tsd_ref_
<<
"."
;
return
true
;
}
...
...
tests/ut/cpp/stub/hccl/hccl_stub.cc
浏览文件 @
9ce5c972
...
...
@@ -103,7 +103,8 @@ hcclResult_t hcom_receive(const char *tag, void *outputPtr, u64 count, hcclDataT
/* 获取梯度参数切分方案 */
hcclResult_t
hcom_get_split_strategy
(
const
char
*
group
,
const
struct
model_feature
*
feature
,
u32
maxSegmentNum
,
u32
*
segmentNum
,
u32
*
segmentIdx
,
GradSplitForceMode
force
)
{
u32
*
segmentNum
,
u32
*
segmentIdx
,
GradSplitForceMode
force
,
OriginalGraphShapeType
shapeType
)
{
return
HCCL_SUCCESS
;
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录