Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
124767b4
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看板
提交
124767b4
编写于
6月 12, 2020
作者:
M
Megvii Engine Team
提交者:
Xu Xinran
6月 19, 2020
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(dnn/fallback): fix mk4_dot test after remove mk4_dot_8x6x4 matmul
GitOrigin-RevId: e3a12cf9b3ad09a133a88472e39cdc753288af92
上级
34659c2e
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
4 addition
and
16 deletion
+4
-16
dnn/src/fallback/matrix_mul/gemm_impl.h
dnn/src/fallback/matrix_mul/gemm_impl.h
+0
-12
dnn/test/arm_common/conv_bias_multi_thread.cpp
dnn/test/arm_common/conv_bias_multi_thread.cpp
+4
-4
未找到文件。
dnn/src/fallback/matrix_mul/gemm_impl.h
浏览文件 @
124767b4
...
...
@@ -52,21 +52,9 @@ class GemmInterleaved<Strategy, true> {
}
size_t
get_b_workspace_size
()
const
{
#if __ARM_FEATURE_DOTPROD
size_t
new_blockn
=
m_strategy
.
block_n
;
if
(
m_strategy
.
KERNEL_W
==
6
&&
m_strategy
.
UNROLL_K
==
4
&&
m_strategy
.
KERNEL_H
==
8
)
{
new_blockn
=
round_up
<
size_t
>
((
m_strategy
.
block_n
-
1
)
%
6
,
4
)
+
m_strategy
.
block_n
/
6
*
6
;
}
size_t
N
=
round_up
(
new_blockn
,
m_strategy
.
KERNEL_W
);
size_t
K
=
round_up
(
m_strategy
.
block_k
,
m_strategy
.
UNROLL_K
);
return
round_up
(
sizeof
(
stype
)
*
N
*
K
,
CACHELINE_SIZE
)
+
m_align_size
;
#else
size_t
N
=
round_up
(
m_strategy
.
block_n
,
m_strategy
.
KERNEL_W
);
size_t
K
=
round_up
(
m_strategy
.
block_k
,
m_strategy
.
UNROLL_K
);
return
round_up
(
sizeof
(
stype
)
*
N
*
K
,
CACHELINE_SIZE
)
+
m_align_size
;
#endif
}
//! temporary storage for output, post process such as add bias or relu will
...
...
dnn/test/arm_common/conv_bias_multi_thread.cpp
浏览文件 @
124767b4
...
...
@@ -1452,7 +1452,7 @@ TEST_F(ARM_COMMON_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_QUANTIZEDSYM_MK4_DOT) {
#if MEGDNN_AARCH64
cb
(
"IM2COLMATMUL:AARCH64_INT8X8X32_MK4_8X12X4_DOTPROD:96"
);
#elif MEGDNN_ARMV7
cb
(
"IM2COLMATMUL:AARCH32_INT8_MK4_8X
6
X4_DOTPROD:96"
);
cb
(
"IM2COLMATMUL:AARCH32_INT8_MK4_8X
4
X4_DOTPROD:96"
);
#endif
#undef cb
}
...
...
@@ -1476,7 +1476,7 @@ TEST_F(ARM_COMMON_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_S8x8x32_MK4_DOT) {
#if MEGDNN_AARCH64
cb
(
"IM2COLMATMUL:AARCH64_INT8X8X32_MK4_8X12X4_DOTPROD:96"
);
#elif MEGDNN_ARMV7
cb
(
"IM2COLMATMUL:AARCH32_INT8_MK4_8X
6
X4_DOTPROD:96"
);
cb
(
"IM2COLMATMUL:AARCH32_INT8_MK4_8X
4
X4_DOTPROD:96"
);
#endif
#undef cb
}
...
...
@@ -1500,7 +1500,7 @@ TEST_F(ARM_COMMON_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_INT8x8x32_MK4_DOT) {
#if MEGDNN_AARCH64
cb
(
"IM2COLMATMUL:AARCH64_INT8X8X32_MK4_8X12X4_DOTPROD:96"
);
#elif MEGDNN_ARMV7
cb
(
"IM2COLMATMUL:AARCH32_INT8_MK4_8X
6
X4_DOTPROD:96"
);
cb
(
"IM2COLMATMUL:AARCH32_INT8_MK4_8X
4
X4_DOTPROD:96"
);
#endif
#undef cb
}
...
...
@@ -1529,7 +1529,7 @@ TEST_F(ARM_COMMON_MULTI_THREADS, CONV_BIAS_CONV1x1_QUANTIZEDSYM_MK4_DOT) {
#if MEGDNN_AARCH64
cb
(
"CONV1x1:AARCH64_INT8X8X32_MK4_8X12X4_DOTPROD"
);
#elif MEGDNN_ARMV7
cb
(
"CONV1x1:AARCH32_INT8_MK4_8X
6
X4_DOTPROD"
);
cb
(
"CONV1x1:AARCH32_INT8_MK4_8X
4
X4_DOTPROD"
);
#endif
#undef cb
}
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录