Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
cc952b2b
MegEngine
项目概览
MegEngine 天元
/
MegEngine
大约 1 年 前同步成功
通知
399
Star
4705
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
cc952b2b
编写于
9月 04, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(rocm): fix rocm megdnntest sleep and a cut code
GitOrigin-RevId: 26de5ca98b28bf93da5c3e9f30c0197eda6d9077
上级
3a03fa7a
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
12 addition
and
1 deletion
+12
-1
dnn/src/rocm/sleep/opr_impl.cpp
dnn/src/rocm/sleep/opr_impl.cpp
+1
-1
src/core/impl/utils/persistent_cache.cpp
src/core/impl/utils/persistent_cache.cpp
+11
-0
未找到文件。
dnn/src/rocm/sleep/opr_impl.cpp
浏览文件 @
cc952b2b
...
@@ -22,7 +22,7 @@ void SleepForwardImpl::exec() {
...
@@ -22,7 +22,7 @@ void SleepForwardImpl::exec() {
double
seconds
=
m_param
.
time
;
double
seconds
=
m_param
.
time
;
megdnn_assert
(
seconds
>
0
);
megdnn_assert
(
seconds
>
0
);
auto
hdl
=
static_cast
<
HandleImpl
*>
(
handle
());
auto
hdl
=
static_cast
<
HandleImpl
*>
(
handle
());
sleep
(
hdl
->
stream
(),
hdl
->
device_prop
().
clockRate
*
1
8.01
*
seconds
);
sleep
(
hdl
->
stream
(),
hdl
->
device_prop
().
clockRate
*
1
000
*
seconds
);
}
}
}
// namespace rocm
}
// namespace rocm
...
...
src/core/impl/utils/persistent_cache.cpp
浏览文件 @
cc952b2b
...
@@ -121,6 +121,17 @@ std::string PersistentCache::make_category_from_comp_node(CompNode comp_node) {
...
@@ -121,6 +121,17 @@ std::string PersistentCache::make_category_from_comp_node(CompNode comp_node) {
prop
.
name
,
prop
.
major
,
prop
.
minor
,
drv
,
cuda_rt
);
prop
.
name
,
prop
.
major
,
prop
.
minor
,
drv
,
cuda_rt
);
break
;
break
;
}
}
#endif
#if MGB_ROCM
case
CompNode
::
DeviceType
::
ROCM
:
{
int
drv
=
-
1
,
hip_rt
=
-
1
;
MGB_ROCM_CHECK
(
hipDriverGetVersion
(
&
drv
));
MGB_ROCM_CHECK
(
hipRuntimeGetVersion
(
&
hip_rt
));
auto
&&
prop
=
env
.
rocm_env
().
device_prop
;
return
ssprintf
(
"plat=rocm;dev=%s;cap=%d.%d,drv=%d;runtime=%d"
,
prop
.
name
,
prop
.
major
,
prop
.
minor
,
drv
,
hip_rt
);
break
;
}
#endif
#endif
case
CompNode
::
DeviceType
::
CPU
:
case
CompNode
::
DeviceType
::
CPU
:
return
"plat=cpu"
;
return
"plat=cpu"
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录