ACLConvertExecutor cannot support dimension > 6, don't let preprocess to add Convert
This commit is contained in:
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)
|
||||
|
||||
Reference in new issue
Block a user