@@ -342,17 +342,16 @@ std::vector<std::string> disabledTestPatterns() {
342
342
retVector.emplace_back (R"( .*smoke_Snippets_MHAWOTransposeEnforceBF16.*)" );
343
343
retVector.emplace_back (R"( .*smoke_Snippets_MHAEnforceBF16.*)" );
344
344
}
345
- if (InferenceEngine::with_cpu_x86_avx512_core ()) {
345
+ // RNN/LSTM/GRU/AUGRU BF16 tests on avx512 core ISA would fail when gemm_avx512 fall back to gemm_avx2.
346
+ if (InferenceEngine::with_cpu_x86_avx512_core ()&& !InferenceEngine::with_cpu_x86_avx512_core_amx_bf16 ()) {
346
347
retVector.emplace_back (R"( smoke.*(AUGRUCellCPUTest|GRUCellCPUTest|LSTMCellLayerCPUTest).CompareWithRefs.*ENFORCE_BF16=YES.*)" );
347
348
retVector.emplace_back (R"( nightly.*bf16.*/(AUGRUSequenceCPUTest|GRUSequenceCPUTest|LSTMSequenceCPUTest|RNNSequenceCPUTest).CompareWithRefs.*ENFORCE_BF16=YES.*)" );
348
- retVector.emplace_back (R"( nightly.*bf16.*/(AUGRUSequenceCPUTest|GRUSequenceCPUTest|LSTMSequenceCPUTest|RNNSequenceCPUTest).CompareWithRefs.*ENFORCE_BF16=YES.*)" );
349
- retVector.emplace_back (R"( .*primitive=jit_gemm.*(ENFORCE_BF16=YES|ENFORCE_BF16, YES).*)" );
350
- retVector.emplace_back (R"( .*(ENFORCE_BF16=YES|ENFORCE_BF16, YES).*(primitive=jit_gemm).*)" );
351
- retVector.emplace_back (R"( smoke_GroupDeconv_2D_AMX_BF16/GroupDeconvolutionLayerCPUTest\.CompareWithRefs/IS=\[\?\.64\.\?\.\?\]_TS=\(\(2\.64\.7\.7\)\)_\(\(2\.64\.5\.7\)\)_\(\(1\.64\.9\.5\)\)_PRC=f32_K=\(3\.3\)_S=\(2\.2\)_PB=\(0\.0\)_PE=\(0\.0\)_D=\(1\.1\)_OP=\(\)_O=64_G=2_AP=explicit_OUT_SH=PARAMETER_OUT_D=\(\(15\.15\)\)_\(\(9\.10\)\)_\(\(19\.9\)\)__inFmts=nhwc_outFmts=nhwc_primitive=jit_avx512_PluginConf_ENFORCE_BF16=YES)" );
352
- retVector.emplace_back (R"( smoke_GroupDeconv_3D_nspc_BF16/GroupDeconvolutionLayerCPUTest\.CompareWithRefs/IS=\[\?\.64\.\?\.\?\.\?\]_TS=\(\(1\.64\.5\.5\.5\)\)_\(\(2\.64\.5\.7\.5\)\)_PRC=f32_K=\(1\.1\.1\)_S=\(2\.2\.2\)_PB=\(0\.0\.0\)_PE=\(0\.0\.0\)_D=\(1\.1\.1\)_OP=\(\)_O=64_G=2_AP=explicit_OUT_SH=PARAMETER_OUT_D=\(\(7\.7\.7\)\)_\(\(7\.9\.7\)\)__inFmts=ndhwc_outFmts=ndhwc_primitive=jit_avx512_PluginConf_ENFORCE_BF16=YES)" );
353
- retVector.emplace_back (R"( smoke_GroupDeconv_3D_nspc_BF16/GroupDeconvolutionLayerCPUTest\.CompareWithRefs/IS=\[\?\.64\.\?\.\?\.\?\]_TS=\(\(1\.64\.5\.5\.5\)\)_\(\(2\.64\.5\.7\.5\)\)_PRC=f32_K=\(3\.3\.3\)_S=\(2\.2\.2\)_PB=\(0\.0\.0\)_PE=\(0\.0\.0\)_D=\(1\.1\.1\)_OP=\(\)_O=64_G=2_AP=explicit_OUT_SH=PARAMETER_OUT_D=\(\(7\.7\.7\)\)_\(\(7\.9\.7\)\)__inFmts=ndhwc_outFmts=ndhwc_primitive=jit_avx512_PluginConf_ENFORCE_BF16=YES)" );
354
- retVector.emplace_back (R"( smoke_GroupDeconv_2D_Blocked_BF16/GroupDeconvolutionLayerCPUTest\.CompareWithRefs/IS=\[\?\.64\.\?\.\?\]_TS=\(\(2\.64\.7\.7\)\)_\(\(2\.64\.5\.7\)\)_\(\(1\.64\.9\.5\)\)_PRC=f32_K=\(3\.3\)_S=\(1\.1\)_PB=\(0\.0\)_PE=\(0\.0\)_D=\(1\.1\)_OP=\(\)_O=64_G=4_AP=explicit_OUT_SH=PARAMETER_OUT_D=\(\(15\.15\)\)_\(\(9\.10\)\)_\(\(19\.9\)\)__inFmts=nChw16c_outFmts=nChw16c_primitive=jit_avx512_PluginConf_ENFORCE_BF16=YES)" );
355
- retVector.emplace_back (R"( smoke_GroupDeconv_2D_AMX_BF16/GroupDeconvolutionLayerCPUTest\.CompareWithRefs/IS=\[\?\.64\.\?\.\?\]_TS=\(\(2\.64\.7\.7\)\)_\(\(2\.64\.5\.7\)\)_\(\(1\.64\.9\.5\)\)_PRC=f32_K=\(1\.1\)_S=\(1\.1\)_PB=\(0\.0\)_PE=\(0\.0\)_D=\(1\.1\)_OP=\(\)_O=64_G=2_AP=explicit_OUT_SH=PARAMETER_OUT_D=\(\(15\.15\)\)_\(\(9\.10\)\)_\(\(19\.9\)\)__inFmts=nhwc_outFmts=nhwc_primitive=jit_avx512_PluginConf_ENFORCE_BF16=YES)" );
349
+ }
350
+ if (InferenceEngine::with_cpu_x86_avx512_core_amx_bf16 ()) {
351
+ // GRUCell and GRUSequence BF16 tests on SPR would fail when gemm_avx512 fall back to gemm_avx2.
352
+ retVector.emplace_back (R"( .*/GRU.*ENFORCE_BF16=YES.*)" );
353
+ // GroupDeconv 3D would fail on BF16 when gemm_avx512 fall back to gemm_avx2.
354
+ retVector.emplace_back (R"( nightly_GroupDeconv_3D_Planar_BF16/GroupDeconvolutionLayerCPUTest\.CompareWithRefs/IS=\[\?.12.\?.\?.\?\]_TS=\(\(2.12.7.7.7\)\)_\(\(2.12.5.7.7\)\)_\(\(1.12.9.4.9\)\)_\(\(2.12.5.7.7\)\)_PRC=f32.*S=\(2.2.2\)_PB=\(0.0.0\)_PE=\(0.0.0\)_D=\(1.1.1\)_OP=\(\)_O=6_G.*primitive=jit_gemm.*_ENFORCE_BF16=YES)" );
356
355
}
357
356
return retVector;
358
357
}
0 commit comments