Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
ccea0e23
MegEngine
项目概览
MegEngine 天元
/
MegEngine
大约 1 年 前同步成功
通知
396
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,发现更多精彩内容 >>
提交
ccea0e23
编写于
4月 27, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(dnn/rdnn): add warmup before profile
GitOrigin-RevId: 7962525e90d80a717b4682f2f666462d49c95b80
上级
58df717e
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
18 addition
and
3 deletion
+18
-3
src/rdnn/impl/algo_chooser.cpp
src/rdnn/impl/algo_chooser.cpp
+1
-1
src/rdnn/impl/profiler.cpp
src/rdnn/impl/profiler.cpp
+17
-2
未找到文件。
src/rdnn/impl/algo_chooser.cpp
浏览文件 @
ccea0e23
...
...
@@ -961,7 +961,7 @@ void AlgoChooser<Opr>::AlgoChooserHelper::profile(
continue
;
})
if
(
!
cur_rst
.
valid
())
{
mgb_log_
warn
(
mgb_log_
debug
(
"timeout when %s; timeout setting: %.3fsec"
,
msg
.
c_str
(),
cur_timeout
);
continue
;
...
...
src/rdnn/impl/profiler.cpp
浏览文件 @
ccea0e23
...
...
@@ -304,6 +304,21 @@ typename TimedProfiler<Opr>::TResult TimedProfiler<Opr>::prof_impl(
RealTimer
timer
;
auto
ev_start
=
cn
.
create_event
(
CompNode
::
Event
::
NEED_TIMER
),
ev_end
=
cn
.
create_event
(
CompNode
::
Event
::
NEED_TIMER
);
for
(
int
i
=
0
;
i
<
5
;
++
i
)
{
if_constexpr
<
opr_supports_preprocess
<
Opr
>
()
>
(
[
&
](
auto
_
)
{
auto
&&
opr
=
_
(
megdnn_opr
);
PreprocessFilter
<
Opr
>*
pf
=
preprocessed_layout
.
empty
()
?
nullptr
:
&
prep_flt
;
APPLY
(
opr
->
exec
(
args
.
as_megdnn
()...,
pf
,
mdn_workspace
),
inp_val
,
out_val
);
},
/* else */
[
&
](
auto
_
)
{
APPLY
(
_
(
megdnn_opr
)
->
exec
(
args
.
as_megdnn
()...,
mdn_workspace
),
inp_val
,
out_val
);
});
}
ev_start
->
record
();
if_constexpr
<
opr_supports_preprocess
<
Opr
>
()
>
(
[
&
](
auto
_
)
{
...
...
@@ -327,12 +342,12 @@ typename TimedProfiler<Opr>::TResult TimedProfiler<Opr>::prof_impl(
while
(
!
ev_end
->
finished
())
{
if
(
timer
.
get_secs
()
>=
next_report_time
)
{
#if MGB_ENABLE_GETENV
mgb_log_
warn
(
mgb_log_
debug
(
"profiling conv algo %s already took %.3f/%.3f secs"
" (limit can be set by MGB_CONV_PROFILING_TIMEOUT) "
,
algo
->
name
(),
timer
.
get_secs
(),
param
.
actual_timeout
);
#else
mgb_log_
warn
(
mgb_log_
debug
(
"profiling conv algo %s already took %.3f/%.3f secs"
,
algo
->
name
(),
timer
.
get_secs
(),
param
.
actual_timeout
);
#endif
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录