Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
e9cc5237
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看板
提交
e9cc5237
编写于
9月 27, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(mgb): format code
GitOrigin-RevId: 3e58539a2f7562c345343cfd5ed3918147b15210
上级
a07fbf79
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
9 addition
and
11 deletion
+9
-11
dnn/src/fallback/convolution/algos.cpp
dnn/src/fallback/convolution/algos.cpp
+4
-4
dnn/src/fallback/convolution/col2img_helper.h
dnn/src/fallback/convolution/col2img_helper.h
+1
-1
dnn/test/fallback/convolution.cpp
dnn/test/fallback/convolution.cpp
+4
-6
未找到文件。
dnn/src/fallback/convolution/algos.cpp
浏览文件 @
e9cc5237
...
...
@@ -586,7 +586,7 @@ bool ConvolutionBackwardDataImpl::AlgoMatrixMul::is_preferred(
}
/* ===================== Matrix mul nchw44 algo ===================== */
namespace
{
namespace
{
void
kern_matmul_nchw44
(
const
NCBKernParam
&
param
)
{
bool
is_xcorr
=
!
param
.
filter_meta
.
should_flip
;
UNPACK_CONV_F32_NCB_KERN_SIZES
(
param
);
...
...
@@ -628,7 +628,7 @@ void kern_matmul_nchw44(const NCBKernParam& param) {
}
TensorND
B_
,
C_
;
for
(
size_t
n
=
0
;
n
<
N
;
++
n
)
{
float
*
C_src
,
*
C_dst
;
float
*
C_src
,
*
C_dst
;
float
*
diff
=
const_cast
<
float
*>
(
param
.
diff
<
float
>
()
+
n
*
param
.
inp_bs
);
float
*
grad
=
param
.
grad
<
float
>
()
+
n
*
param
.
out_bs
;
if
(
is1X1
)
{
...
...
@@ -637,13 +637,13 @@ void kern_matmul_nchw44(const NCBKernParam& param) {
C_src
=
static_cast
<
float
*>
(
bundle
.
get
(
0
));
}
{
B_
.
layout
=
TensorLayout
({
OC
/
4
,
IH
*
IW
,
4
},
param
.
diff_type
);
B_
.
layout
=
TensorLayout
({
OC
/
4
,
IH
*
IW
,
4
},
param
.
diff_type
);
B_
.
reset_ptr
(
static_cast
<
void
*>
(
diff
));
C_
.
layout
=
TensorLayout
({
IC
/
4
*
FH
*
FW
,
IH
*
IW
,
4
},
param
.
grad_type
);
C_
.
reset_ptr
(
C_src
);
Workspace
workspace
(
static_cast
<
dt_byte
*>
(
bundle
.
get
(
1
)),
bundle
.
get_size
(
1
));
auto
matmul_opr
=
get_matmul_opr
(
param
);
auto
matmul_opr
=
get_matmul_opr
(
param
);
matmul_opr
->
exec
(
A_dst
,
B_
,
C_
,
workspace
);
}
...
...
dnn/src/fallback/convolution/col2img_helper.h
浏览文件 @
e9cc5237
...
...
@@ -86,7 +86,7 @@ void col2img_stride_padding_nchw44(
if
(
h
>=
0
&&
h
<
OH
&&
w
>=
0
&&
w
<
OW
)
{
float
*
dst_ptr
=
dst
+
(
ic
*
OH
*
OW
+
h
*
OW
+
w
)
*
4
;
GI_FLOAT32_t
dst_data
=
GiLoadFloat32
(
dst_ptr
);
GI_FLOAT32_t
src_data
=
GiLoadFloat32
(
src
+
i
);
GI_FLOAT32_t
src_data
=
GiLoadFloat32
(
src
+
i
);
GiStoreFloat32
(
dst_ptr
,
GiAddFloat32
(
dst_data
,
src_data
));
}
i
+=
4
;
...
...
dnn/test/fallback/convolution.cpp
浏览文件 @
e9cc5237
...
...
@@ -804,13 +804,11 @@ TEST_F(FALLBACK, BENCHMARK_CONVOLUTION_BACKWARD_DATA_NCHW44) {
.
set_dtype
(
2
,
dtype
::
Float32
{})
.
set_times
(
RUN
);
auto
tnchw
=
benchmarker_fallback
.
set_param
(
param
)
.
exec
(
TensorLayoutArray
{
filter
,
diff
,
grad
});
auto
tnchw
=
benchmarker_fallback
.
set_param
(
param
).
exec
(
TensorLayoutArray
{
filter
,
diff
,
grad
});
param
.
format
=
Param
::
Format
::
NCHW44
;
auto
tnchw44
=
benchmarker_fallback
.
set_param
(
param
)
.
exec
(
TensorLayoutArray
{
filter_nchw44
,
diff_nchw44
,
grad_nchw44
});
auto
tnchw44
=
benchmarker_fallback
.
set_param
(
param
).
exec
(
TensorLayoutArray
{
filter_nchw44
,
diff_nchw44
,
grad_nchw44
});
size_t
IC
=
ic
;
size_t
FH
=
fh
;
size_t
FW
=
fw
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录