Browse Source

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

GitOrigin-RevId: cc7283c6a2
tags/v1.7.0.m1
Megvii Engine Team 3 years ago
parent
commit
02d5f46d90
1 changed files with 3 additions and 3 deletions
  1. +3
    -3
      dnn/test/x86/conv_bias.cpp

+ 3
- 3
dnn/test/x86/conv_bias.cpp View File

@@ -935,7 +935,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_INT8X8X32) {

#if MEGDNN_X86_WITH_MKL_DNN
if (megdnn::x86::is_supported(x86::SIMDType::VNNI)) {
cb("IM2COLMATMUL:X86_INT8X8X32_MKLDNN");
cb("IM2COLMATMUL");
}
#endif
#if MEGDNN_X86_WITH_VNNI
@@ -1647,7 +1647,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_QINT8) {

#if MEGDNN_X86_WITH_MKL_DNN
if (x86::is_supported(x86::SIMDType::VNNI)) {
cb("IM2COLMATMUL:X86_INT8X8X32_MKLDNN");
cb("IM2COLMATMUL");
}
#endif
#if MEGDNN_X86_WITH_VNNI
@@ -1718,7 +1718,7 @@ TEST_F(X86_MULTI_THREADS, CONV_BIAS_IM2COLMATMUL_QINT8_FILTER_PREPROCESS) {

#if MEGDNN_X86_WITH_MKL_DNN
if (x86::is_supported(x86::SIMDType::VNNI)) {
cb("IM2COLMATMUL:X86_INT8X8X32_MKLDNN");
cb("IM2COLMATMUL");
}
#endif
#if MEGDNN_X86_WITH_VNNI


Loading…
Cancel
Save