reduce 6d+ fix (#16931)
This commit is contained in:
parent
df5ada8b19
commit
fe98b8ee13
@ -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;
|
||||
}
|
||||
|
||||
|
@ -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;
|
||||
|
@ -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.*)");
|
||||
|
Loading…
Reference in New Issue
Block a user