Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
2301_77200941
mindspore
提交
50879454
M
mindspore
项目概览
2301_77200941
/
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
50879454
编写于
7月 28, 2020
作者:
X
xsmq
提交者:
Gitee
7月 28, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
回退 'Pull Request !3236 : use akg to compile gpu ops rather than _akg'
上级
46ae7879
变更
6
隐藏空白更改
内联
并排
Showing
6 changed file
with
7 addition
and
7 deletion
+7
-7
CMakeLists.txt
CMakeLists.txt
+1
-1
akg
akg
+1
-1
build.sh
build.sh
+3
-3
cmake/package.cmake
cmake/package.cmake
+1
-1
mindspore/ccsrc/backend/kernel_compiler/common_utils.h
mindspore/ccsrc/backend/kernel_compiler/common_utils.h
+1
-1
model_zoo/official/cv/maskrcnn/scripts/run_standalone_train.sh
..._zoo/official/cv/maskrcnn/scripts/run_standalone_train.sh
+0
-0
未找到文件。
CMakeLists.txt
浏览文件 @
50879454
...
...
@@ -42,7 +42,7 @@ if (NOT Patch_FOUND)
endif
()
message
(
PATCH_EXECUTABLE =
${
Patch_EXECUTABLE
}
)
if
(
ENABLE_AKG
AND
(
ENABLE_D OR ENABLE_GPU
)
)
if
(
ENABLE_AKG AND
ENABLE_D
)
add_subdirectory
(
"
${
CMAKE_SOURCE_DIR
}
/akg"
)
endif
()
...
...
akg
@
f60af9df
Subproject commit
5237a9a08bfa03cc4c6be912a43d5efc0f4f153d
Subproject commit
f60af9df4220bf3db5de2b224418953c0dc1f625
build.sh
浏览文件 @
50879454
...
...
@@ -285,7 +285,7 @@ fi
echo
"---------------- MindSpore: build start ----------------"
mkdir
-pv
"
${
BUILD_PATH
}
/package/mindspore/lib"
git submodule update
--init
graphengine
if
[[
"X
$ENABLE_AKG
"
=
"Xon"
]]
&&
[[
"X
$ENABLE_D
"
=
"Xon"
||
"X
$ENABLE_GPU
"
=
"Xon"
]]
;
then
if
[[
"X
$ENABLE_AKG
"
=
"Xon"
]]
&&
[[
"X
$ENABLE_D
"
=
"Xon"
]]
;
then
git submodule update
--init
--recursive
akg
fi
...
...
@@ -340,7 +340,7 @@ build_mindspore()
CMAKE_ARGS
=
"
${
CMAKE_ARGS
}
-DENABLE_D=ON"
fi
if
[[
"X
$ENABLE_GPU
"
=
"Xon"
]]
;
then
CMAKE_ARGS
=
"
${
CMAKE_ARGS
}
-DENABLE_GPU=ON -D
USE_CUDA=ON -DUSE_CUDNN=ON -D
CUDA_PATH=
$CUDA_PATH
-DCUDNN_PATH=
$CUDNN_PATH
-DMS_REQUIRE_CUDA_VERSION=
${
CUDA_VERSION
}
"
CMAKE_ARGS
=
"
${
CMAKE_ARGS
}
-DENABLE_GPU=ON -DCUDA_PATH=
$CUDA_PATH
-DCUDNN_PATH=
$CUDNN_PATH
-DMS_REQUIRE_CUDA_VERSION=
${
CUDA_VERSION
}
"
fi
if
[[
"X
$ENABLE_CPU
"
=
"Xon"
]]
;
then
CMAKE_ARGS
=
"
${
CMAKE_ARGS
}
-DENABLE_CPU=ON"
...
...
@@ -351,7 +351,7 @@ build_mindspore()
if
[[
"X
$USE_GLOG
"
=
"Xon"
]]
;
then
CMAKE_ARGS
=
"
${
CMAKE_ARGS
}
-DUSE_GLOG=ON"
fi
if
[[
"X
$ENABLE_AKG
"
=
"Xon"
]]
&&
[[
"X
$ENABLE_D
"
=
"Xon"
||
"X
$ENABLE_GPU
"
=
"Xon"
]]
;
then
if
[[
"X
$ENABLE_AKG
"
=
"Xon"
]]
&&
[[
"X
$ENABLE_D
"
=
"Xon"
]]
;
then
CMAKE_ARGS
=
"
${
CMAKE_ARGS
}
-DENABLE_AKG=ON"
fi
if
[[
"X
$ENABLE_SERVING
"
=
"Xon"
]]
;
then
...
...
cmake/package.cmake
浏览文件 @
50879454
...
...
@@ -253,7 +253,7 @@ if (ENABLE_GPU)
endif
()
endif
()
if
(
ENABLE_
AKG
AND
(
ENABLE_D OR ENABLE_GPU
)
)
if
(
ENABLE_
D AND ENABLE_AKG
)
set
(
AKG_PATH
${
CMAKE_SOURCE_DIR
}
/build/mindspore/akg
)
install
(
DIRECTORY
...
...
mindspore/ccsrc/backend/kernel_compiler/common_utils.h
浏览文件 @
50879454
...
...
@@ -40,7 +40,7 @@ constexpr auto kProcessorCuda = "cuda";
constexpr
auto
kJsonSuffix
=
".json"
;
constexpr
auto
kInfoSuffix
=
".info"
;
constexpr
unsigned
int
AUTODIFF_COMPILE_OVERTIME
=
600
;
constexpr
auto
kAkgModule
=
"
akg.ms
"
;
constexpr
auto
kAkgModule
=
"
_akg
"
;
constexpr
auto
kArgDataformat
=
"data_format"
;
const
std
::
vector
<
std
::
string
>
support_devices
=
{
"aicore"
,
"aicpu"
,
"cuda"
};
...
...
model_zoo/official/cv/maskrcnn/scripts/run_standalone_train.sh
100755 → 100644
浏览文件 @
50879454
文件模式从 100755 更改为 100644
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录