Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
ed04b8e1
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看板
提交
ed04b8e1
编写于
4月 06, 2020
作者:
Z
zhoufeng
提交者:
高东海
4月 10, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
default build command
"-z" (minddata) and "-M on" (gpu) by default in build.sh
上级
d84cfb01
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
21 addition
and
11 deletion
+21
-11
build.sh
build.sh
+17
-7
cmake/mind_expression.cmake
cmake/mind_expression.cmake
+4
-4
未找到文件。
build.sh
浏览文件 @
ed04b8e1
...
...
@@ -26,7 +26,7 @@ usage()
echo
"Usage:"
echo
"bash build.sh [-d] [-r] [-v] [-c on|off] [-t on|off] [-g on|off] [-h] [-s] [-b ge|cpu] [-m infer|train]
\\
"
echo
" [-a on|off] [-g on|off] [-p on|off] [-i] [-L] [-R] [-D on|off] [-j[n]] [-e gpu|d|cpu]
\\
"
echo
" [-P on|off] [-z] [-M on|off] [-V 9.2|10.1] [-I] [-K]"
echo
" [-P on|off] [-z
[on|off]
] [-M on|off] [-V 9.2|10.1] [-I] [-K]"
echo
""
echo
"Options:"
echo
" -d Debug mode"
...
...
@@ -50,8 +50,8 @@ usage()
echo
" -P Enable dump anf graph to file in ProtoBuffer format, default on"
echo
" -Q Enable dump end to end, default off"
echo
" -D Enable dumping of function graph ir, default on"
echo
" -z Compile dataset & mindrecord, default o
ff
"
echo
" -M Enable MPI and NCCL for GPU training, default o
ff
"
echo
" -z Compile dataset & mindrecord, default o
n
"
echo
" -M Enable MPI and NCCL for GPU training, default o
n
"
echo
" -V Specify the minimum required cuda version, default CUDA 9.2"
echo
" -I Compile predict, default off"
echo
" -K Compile with AKG, default off"
...
...
@@ -88,8 +88,8 @@ checkopts()
ENABLE_DUMP2PROTO
=
"on"
ENABLE_DUMPE2E
=
"off"
ENABLE_DUMP_IR
=
"on"
COMPILE_MINDDATA
=
"o
ff
"
ENABLE_MPI
=
"o
ff
"
COMPILE_MINDDATA
=
"o
n
"
ENABLE_MPI
=
"o
n
"
CUDA_VERSION
=
"9.2"
COMPILE_PREDICT
=
"off"
USE_GLOG
=
"on"
...
...
@@ -177,7 +177,7 @@ checkopts()
if
[[
"X
$OPTARG
"
==
"Xgpu"
]]
;
then
ENABLE_GPU
=
"on"
ENABLE_CPU
=
"on"
elif
[[
"X
$OPTARG
"
==
"Xd"
]]
;
then
elif
[[
"X
$OPTARG
"
==
"Xd"
||
"X
$OPTARG
"
==
"Xascend"
]]
;
then
ENABLE_D
=
"on"
ENABLE_CPU
=
"on"
elif
[[
"X
$OPTARG
"
==
"Xcpu"
]]
;
then
...
...
@@ -216,7 +216,17 @@ checkopts()
echo
"enable dump function graph ir"
;;
z
)
COMPILE_MINDDATA
=
"on"
eval
ARG
=
\$\{
$OPTIND
\}
if
[[
-n
$ARG
&&
$ARG
!=
-
*
]]
;
then
OPTARG
=
$ARG
check_on_off
$OPTARG
z
OPTIND
=
$((
OPTIND
+
1
))
else
OPTARG
=
""
fi
if
[[
"X
$OPTARG
"
==
"Xoff"
]]
;
then
COMPILE_MINDDATA
=
"off"
fi
;;
I
)
COMPILE_PREDICT
=
"on"
...
...
cmake/mind_expression.cmake
浏览文件 @
ed04b8e1
...
...
@@ -29,11 +29,11 @@ if (ENABLE_GPU)
include
(
${
CMAKE_SOURCE_DIR
}
/cmake/external_libs/dmlc_core.cmake
)
include
(
${
CMAKE_SOURCE_DIR
}
/cmake/external_libs/rang.cmake
)
include
(
${
CMAKE_SOURCE_DIR
}
/cmake/external_libs/tvm_gpu.cmake
)
endif
()
if
(
ENABLE_MPI
)
include
(
${
CMAKE_SOURCE_DIR
}
/cmake/external_libs/nccl.cmake
)
include
(
${
CMAKE_SOURCE_DIR
}
/cmake/external_libs/ompi.cmake
)
if
(
ENABLE_MPI
)
include
(
${
CMAKE_SOURCE_DIR
}
/cmake/external_libs/nccl.cmake
)
include
(
${
CMAKE_SOURCE_DIR
}
/cmake/external_libs/ompi.cmake
)
endif
()
endif
()
if
(
ENABLE_GE
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录