Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
ccd933e6
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
410
Star
4707
Fork
583
代码
文件
提交
分支
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看板
提交
ccd933e6
编写于
2月 28, 2023
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(dnn/fallback): fix gi f43 winograd algo error
GitOrigin-RevId: 99e6c498b97030f69dc9f390b1ce8883555f0267
上级
2b254d61
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
13 addition
and
13 deletion
+13
-13
dnn/src/fallback/conv_bias/gi/fp16/strategy_f43_mk8_nchw88.cpp
...rc/fallback/conv_bias/gi/fp16/strategy_f43_mk8_nchw88.cpp
+3
-2
dnn/test/common/conv_bias.cpp
dnn/test/common/conv_bias.cpp
+3
-5
dnn/test/fallback/conv_bias.cpp
dnn/test/fallback/conv_bias.cpp
+7
-6
未找到文件。
dnn/src/fallback/conv_bias/gi/fp16/strategy_f43_mk8_nchw88.cpp
浏览文件 @
ccd933e6
...
...
@@ -439,8 +439,9 @@ struct OutputTransformF43_NCHW88 {
UNROLL_CALL_NOWRAPPER_D2
(
5
,
6
,
cb
);
#undef cb
const
gi_float16_t
*
buf_base
=
output_transform_buf
+
ocb
*
nr_units_in_tile
*
4
+
unit_idx
*
4
;
const
gi_float16_t
*
buf_base
=
output_transform_buf
+
ocb
*
nr_units_in_tile
*
pack_size
+
unit_idx
*
pack_size
;
const
gi_float16_t
*
buf_ptr
=
nullptr
;
// load line 1 -> v10 ... v15
...
...
dnn/test/common/conv_bias.cpp
浏览文件 @
ccd933e6
...
...
@@ -1743,11 +1743,9 @@ std::vector<conv_bias::TestArg> get_nchw88_conv_bias_args(
if
(
ic
%
(
group
*
8
)
||
oc
%
(
group
*
8
))
{
continue
;
}
if
(
kernel
<
h
||
kernel
<
w
)
{
continue
;
}
pack
(
n
,
oc
,
ic
,
h
,
w
,
kernel
,
stride
,
pad
,
group
,
nlmode
,
bias
);
if
(
kernel
<
h
&&
kernel
<
w
)
pack
(
n
,
oc
,
ic
,
h
,
w
,
kernel
,
stride
,
pad
,
group
,
nlmode
,
bias
);
}
}
return
args
;
...
...
dnn/test/fallback/conv_bias.cpp
浏览文件 @
ccd933e6
...
...
@@ -609,6 +609,7 @@ TEST_F(FALLBACK_MULTI_THREADS, CONVBIAS_GI_WINOGRAD_F43_4_NCHW44) {
}
#if defined(GI_SUPPORT_F16)
TEST_F
(
FALLBACK_MULTI_THREADS
,
CONVBIAS_GI_WINOGRAD_F23_8_NCHW88_FP16
)
{
using
namespace
conv_bias
;
std
::
vector
<
TestArg
>
args
=
...
...
@@ -616,9 +617,9 @@ TEST_F(FALLBACK_MULTI_THREADS, CONVBIAS_GI_WINOGRAD_F23_8_NCHW88_FP16) {
Checker
<
ConvBiasForward
,
OprWeightPreprocessProxy
<
ConvBiasForward
>>
checker
(
handle
());
Float16PeriodicalRNG
rng
(
0x3c00
);
Float16PeriodicalRNG
rng
(
0x3c00
);
// (-1.0, 1.0)
check_winograd_fp16
(
"8:2:"
,
checker
,
args
,
&
rng
,
0.00
3
,
param
::
MatrixMul
::
Format
::
MK8
,
"8:2:"
,
checker
,
args
,
&
rng
,
0.00
9
,
param
::
MatrixMul
::
Format
::
MK8
,
"WINOGRAD_NCHW88"
);
}
...
...
@@ -629,9 +630,9 @@ TEST_F(FALLBACK_MULTI_THREADS, CONVBIAS_GI_WINOGRAD_F43_8_NCHW88_FP16) {
Checker
<
ConvBiasForward
,
OprWeightPreprocessProxy
<
ConvBiasForward
>>
checker
(
handle
());
Float16PeriodicalRNG
rng
(
0x3
c00
);
Float16PeriodicalRNG
rng
(
0x3
800
);
// (-0.5, 0.5)
check_winograd_fp16
(
"8:4:"
,
checker
,
args
,
&
rng
,
0.0
06
,
param
::
MatrixMul
::
Format
::
MK8
,
"8:4:"
,
checker
,
args
,
&
rng
,
0.0
27
,
param
::
MatrixMul
::
Format
::
MK8
,
"WINOGRAD_NCHW88"
);
}
...
...
@@ -642,9 +643,9 @@ TEST_F(FALLBACK_MULTI_THREADS, CONVBIAS_GI_WINOGRAD_F63_8_NCHW88_FP16) {
Checker
<
ConvBiasForward
,
OprWeightPreprocessProxy
<
ConvBiasForward
>>
checker
(
handle
());
Float16PeriodicalRNG
rng
(
0x3
c00
);
Float16PeriodicalRNG
rng
(
0x3
800
);
// (-0.5, 0.5)
check_winograd_fp16
(
"8:6:"
,
checker
,
args
,
&
rng
,
0.0
19
,
param
::
MatrixMul
::
Format
::
MK8
,
"8:6:"
,
checker
,
args
,
&
rng
,
0.0
6
,
param
::
MatrixMul
::
Format
::
MK8
,
"WINOGRAD_NCHW88"
);
}
#endif
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录