reduce 6d+ fix (#16931)

This commit is contained in:
Aleksandr Voron 2023-04-14 20:09:22 +02:00 committed by GitHub
parent df5ada8b19
commit fe98b8ee13
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 10 additions and 1 deletions

View File

@ -63,6 +63,11 @@ public:
return false;
}
}
if (srcDescs[0]->getShape().getRank() >= arm_compute::MAX_DIMS) {
DEBUG_LOG("ACL supports ", arm_compute::MAX_DIMS,
" dimensions maximum. src[0] shape rank is ", srcDescs[0]->getShape().getRank());
return false;
}
return true;
}

View File

@ -2023,7 +2023,11 @@ void Reduce::createPrimitive() {
jcp.layout = layout;
jcp.reduce_mode = getAlgorithm();
#if defined(OPENVINO_ARCH_X86_64)
compile_post_kernel = true;
#else
compile_post_kernel = false;
#endif // OPENVINO_ARCH_X86_64
if (mayiuse(cpu::x64::avx512_core)) {
blk_size = 16;

View File

@ -236,7 +236,7 @@ std::vector<std::string> disabledTestPatterns() {
retVector.emplace_back(R"(smoke_TrivialLoop/TrivialLoopTest.AutoSlicingInputWithDynCondition_(CheckPredefinedValues|CheckReference).*)");
retVector.emplace_back(R"(smoke_LSTMSequenceCommonZeroClip/LSTMSequenceTest.CompareWithRefs.*)");
retVector.emplace_back(R"(smoke_LSTMSequenceCommonClip/LSTMSequenceTest.CompareWithRefs.*)");
retVector.emplace_back(R"(smoke_Decomposition_(3|4|10)D/Mvn6LayerTest.CompareWithRefs.*)");
retVector.emplace_back(R"(smoke_Decomposition_(3|4)D/Mvn6LayerTest.CompareWithRefs.*)");
retVector.emplace_back(R"(smoke_AvgPool_ExplicitPad_CeilRounding/PoolingLayerTest.CompareWithRefs.*)");
retVector.emplace_back(R"(smoke_ReduceLogicalOneAxis/ReduceOpsLayerTest.CompareWithRefs.*)");
retVector.emplace_back(R"(smoke_Reduce_InputShapes/ReduceOpsLayerTest.CompareWithRefs.*)");