reduce 6d+ fix (#16931)
This commit is contained in:
parent
df5ada8b19
commit
fe98b8ee13
@ -63,6 +63,11 @@ public:
|
|||||||
return false;
|
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;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -2023,7 +2023,11 @@ void Reduce::createPrimitive() {
|
|||||||
jcp.layout = layout;
|
jcp.layout = layout;
|
||||||
jcp.reduce_mode = getAlgorithm();
|
jcp.reduce_mode = getAlgorithm();
|
||||||
|
|
||||||
|
#if defined(OPENVINO_ARCH_X86_64)
|
||||||
compile_post_kernel = true;
|
compile_post_kernel = true;
|
||||||
|
#else
|
||||||
|
compile_post_kernel = false;
|
||||||
|
#endif // OPENVINO_ARCH_X86_64
|
||||||
|
|
||||||
if (mayiuse(cpu::x64::avx512_core)) {
|
if (mayiuse(cpu::x64::avx512_core)) {
|
||||||
blk_size = 16;
|
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_TrivialLoop/TrivialLoopTest.AutoSlicingInputWithDynCondition_(CheckPredefinedValues|CheckReference).*)");
|
||||||
retVector.emplace_back(R"(smoke_LSTMSequenceCommonZeroClip/LSTMSequenceTest.CompareWithRefs.*)");
|
retVector.emplace_back(R"(smoke_LSTMSequenceCommonZeroClip/LSTMSequenceTest.CompareWithRefs.*)");
|
||||||
retVector.emplace_back(R"(smoke_LSTMSequenceCommonClip/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_AvgPool_ExplicitPad_CeilRounding/PoolingLayerTest.CompareWithRefs.*)");
|
||||||
retVector.emplace_back(R"(smoke_ReduceLogicalOneAxis/ReduceOpsLayerTest.CompareWithRefs.*)");
|
retVector.emplace_back(R"(smoke_ReduceLogicalOneAxis/ReduceOpsLayerTest.CompareWithRefs.*)");
|
||||||
retVector.emplace_back(R"(smoke_Reduce_InputShapes/ReduceOpsLayerTest.CompareWithRefs.*)");
|
retVector.emplace_back(R"(smoke_Reduce_InputShapes/ReduceOpsLayerTest.CompareWithRefs.*)");
|
||||||
|
Loading…
Reference in New Issue
Block a user