ACLConvertExecutor cannot support dimension > 6, don't let preprocess to add Convert

This commit is contained in:
River.Li committed 2023-07-17 03:59:28 +08:00
1 parent 516e644955
commit fd7a8b35af
2 files changed
+7 -4

No files matched your search

@@ -30,9 +30,11 @@ bool ov::pass::AddPreprocessing::run_on_model(const std::shared_ptr<ov::Model>&
auto& legacy_preproc = input_info->getPreProcess();
#if !defined(OPENVINO_ARCH_ARM64) && !defined(OPENVINO_ARCH_ARM)
// ACLConvertExecutor cannot support dimension > 6, don't change precision to avoid adding Convert
preproc.input(i).tensor().set_element_type(
InferenceEngine::details::convertPrecision(input_info->getPrecision()));
#endif
if (input_info->getLayout() != InferenceEngine::Layout::BLOCKED &&
input_info->getLayout() != InferenceEngine::Layout::SCALAR) {
std::stringstream stream;
@@ -118,11 +120,14 @@ bool ov::pass::AddPreprocessing::run_on_model(const std::shared_ptr<ov::Model>&
// ExecutableNetwork
ov::legacy_convert::fill_output_info(const_output, output_info);
OPENVINO_ASSERT(output_info);
#if !defined(OPENVINO_ARCH_ARM64) && !defined(OPENVINO_ARCH_ARM)
// ACLConvertExecutor cannot support dimension > 6, don't change precision to avoid adding Convert
auto element_type = InferenceEngine::details::convertPrecision(output_info->getPrecision());
if (element_type != model->output(i).get_element_type()) {
preproc.output(i).tensor().set_element_type(element_type);
}
#endif
if (output_info->getLayout() != InferenceEngine::Layout::BLOCKED &&
output_info->getLayout() != InferenceEngine::Layout::SCALAR) {
std::stringstream stream;
@@ -234,8 +234,6 @@ std::vector<std::string> disabledTestPatterns() {
retVector.emplace_back(R"(smoke_NegativeQuantizedMatMulMultiplyFusion.*)");
// int8 specific
retVector.emplace_back(R"(smoke_Quantized.*)");
// ACL convert doesn't support dimension > 6, this test shape_size = 10
retVector.emplace_back(R"(smoke_TestNumpyBroadcastNgraphEvaluate/BroadcastLayerTest.CompareWithRefs.*_inNPrec=(U8|I8)_trgDev=CPU.*)");
#endif
#if defined(OPENVINO_ARCH_ARM)