Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
97beae2f
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
403
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看板
提交
97beae2f
编写于
3月 02, 2021
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(megdnn): fix megdnn benchmark testcase
GitOrigin-RevId: 726971474a8b056b7bb42004ae1a89a054562ae6
上级
1d4baa42
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
12 addition
and
4 deletion
+12
-4
dnn/test/cuda/batch_conv_bias.cpp
dnn/test/cuda/batch_conv_bias.cpp
+8
-4
dnn/test/cuda/benchmark.cpp
dnn/test/cuda/benchmark.cpp
+4
-0
未找到文件。
dnn/test/cuda/batch_conv_bias.cpp
浏览文件 @
97beae2f
...
...
@@ -241,10 +241,14 @@ void benchmark_target_algo(Handle* handle, const std::vector<BenchArgs>& args,
"v" V(CUDNN_MAJOR) "." V(CUDNN_MINOR) "." V(CUDNN_PATCHLEVEL)
benchmarker_cudnn
.
set_before_exec_callback
(
conv_bias
::
ConvBiasAlgoChecker
<
ConvBiasForward
>
(
"CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_"
"GEMM"
CUDNN_VERSION_STRING
));
benchmarker_matmul
.
set_before_exec_callback
(
AlgoChecker
<
BatchedMatrixMul
>
(
"BRUTE_FORCE-CUBLAS"
));
ConvBiasForward
::
algo_name
<
ConvBias
::
DefaultParam
>
(
"CUDNN:ConvBiasActivation:"
"CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_PRECOMP_"
"GEMM"
CUDNN_VERSION_STRING
,
{})
.
c_str
()));
benchmarker_matmul
.
set_before_exec_callback
(
AlgoChecker
<
BatchedMatrixMul
>
(
ExecutionPolicyAlgoName
{
"BRUTE_FORCE"
,
{{
"CUBLAS"
,
{}}}}));
benchmarker
.
set_dtype
(
0
,
src_dtype
)
.
set_dtype
(
1
,
filter_dtype
)
...
...
dnn/test/cuda/benchmark.cpp
浏览文件 @
97beae2f
...
...
@@ -41,10 +41,12 @@ TEST_F(CUDA, BENCHMARK_CONVOLUTION_8X8X32)
auto
time_in_ms_float
=
benchmarker
.
set_param
(
param_float
)
.
set_dtype
(
0
,
dtype
::
Float32
())
.
set_dtype
(
1
,
dtype
::
Float32
())
.
set_dtype
(
2
,
dtype
::
Float32
())
.
execs
({
src_float
,
filter_float
,
{}});
auto
time_in_ms_int
=
benchmarker
.
set_param
(
param_int
)
.
set_dtype
(
0
,
dtype
::
Int8
())
.
set_dtype
(
1
,
dtype
::
Int8
())
.
set_dtype
(
2
,
dtype
::
Int32
())
.
execs
({
src_int
,
filter_int
,
{}});
std
::
cout
<<
"1x1: N="
<<
N
<<
" OC="
<<
OC
<<
" IC="
<<
IC
<<
" H="
<<
H
<<
" W="
<<
W
...
...
@@ -67,10 +69,12 @@ TEST_F(CUDA, BENCHMARK_CONVOLUTION_8X8X32)
auto
time_in_ms_float
=
benchmarker
.
set_param
(
param_float
)
.
set_dtype
(
0
,
dtype
::
Float32
())
.
set_dtype
(
1
,
dtype
::
Float32
())
.
set_dtype
(
2
,
dtype
::
Float32
())
.
execs
({
src_float
,
filter_float
,
{}});
auto
time_in_ms_int
=
benchmarker
.
set_param
(
param_int
)
.
set_dtype
(
0
,
dtype
::
Int8
())
.
set_dtype
(
1
,
dtype
::
Int8
())
.
set_dtype
(
2
,
dtype
::
Int32
())
.
execs
({
src_int
,
filter_int
,
{}});
std
::
cout
<<
"chanwise: N="
<<
N
<<
" C="
<<
C
<<
" H="
<<
H
<<
" W="
<<
W
<<
" F="
<<
F
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录