Skip to content

MegEngine
MegEngine

MegEngine 天元 / MegEngine
1 年多 前同步成功

通知 404
Star 4705
Fork 582
MegEngine
MegEngine
提交 02d5f46d 编写于 作者: M Megvii Engine Team
浏览文件
操作

fix(mgb/x86): fix convbias crash on X86 

GitOrigin-RevId: cc7283c6a26cd35445594d2ff2d8e5567f61142b
上级 accb2d8d
隐藏空白更改
内联 并排
Showing with 3 addition and 3 deletion
dnn/test/x86/conv_bias.cpp
浏览文件 @ 02d5f46d
...@@ -935,7 +935,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_INT8X8X32) { ...@@ -935,7 +935,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_INT8X8X32) {
#if MEGDNN_X86_WITH_MKL_DNN #if MEGDNN_X86_WITH_MKL_DNN
if (megdnn::x86::is_supported(x86::SIMDType::VNNI)) { if (megdnn::x86::is_supported(x86::SIMDType::VNNI)) {
cb("IM2COLMATMUL:X86_INT8X8X32_MKLDNN"); cb("IM2COLMATMUL");
} }
#endif #endif
#if MEGDNN_X86_WITH_VNNI #if MEGDNN_X86_WITH_VNNI
...@@ -1647,7 +1647,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_QINT8) { ...@@ -1647,7 +1647,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_QINT8) {
#if MEGDNN_X86_WITH_MKL_DNN #if MEGDNN_X86_WITH_MKL_DNN
if (x86::is_supported(x86::SIMDType::VNNI)) { if (x86::is_supported(x86::SIMDType::VNNI)) {
cb("IM2COLMATMUL:X86_INT8X8X32_MKLDNN"); cb("IM2COLMATMUL");
} }
#endif #endif
#if MEGDNN_X86_WITH_VNNI #if MEGDNN_X86_WITH_VNNI
...@@ -1718,7 +1718,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_QINT8_FILTER_PREPROCESS) { ...@@ -1718,7 +1718,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_QINT8_FILTER_PREPROCESS) {
#if MEGDNN_X86_WITH_MKL_DNN #if MEGDNN_X86_WITH_MKL_DNN
if (x86::is_supported(x86::SIMDType::VNNI)) { if (x86::is_supported(x86::SIMDType::VNNI)) {
cb("IM2COLMATMUL:X86_INT8X8X32_MKLDNN"); cb("IM2COLMATMUL");
} }
#endif #endif
#if MEGDNN_X86_WITH_VNNI #if MEGDNN_X86_WITH_VNNI
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或