Update Reorder::reorderData

riverlijunjie · Sep 13, 2023 · 8b164ce · 8b164ce
1 parent a3fc7b9
commit 8b164ce
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/plugins/intel_cpu/src/nodes/reorder.cpp b/src/plugins/intel_cpu/src/nodes/reorder.cpp
@@ -316,7 +316,7 @@ std::string Reorder::getReorderArgs(const MemoryDesc &parentDesc, const MemoryDe
     return inArgs + "_" + outArgs;
 }
 
-void Reorder::reorderData(const IMemory &input, const IMemory &output, MultiCachePtr cache) {
+void Reorder::reorderData2(const IMemory &input, const IMemory &output, MultiCachePtr cache) {
     if (!input.getDesc().isDefined() || !output.getDesc().isDefined())
         IE_THROW() << "Can't reorder data with dynamic shapes";
 
@@ -372,7 +372,7 @@ void Reorder::reorderData(const IMemory &input, const IMemory &output, MultiCach
     }
 }
 
-void Reorder::reorderData2(const IMemory& input, const IMemory& output, MultiCachePtr cache) {
+void Reorder::reorderData(const IMemory& input, const IMemory& output, MultiCachePtr cache) {
     if (!input.getDesc().isDefined() || !output.getDesc().isDefined())
         IE_THROW() << "Can't reorder data with dynamic shapes";
 
@@ -396,7 +396,9 @@ void Reorder::reorderData2(const IMemory& input, const IMemory& output, MultiCac
                                                           src_permutation);
         executor->prepareParams(engine, cache, static_cast<MemoryCPtr>(&input), static_cast<MemoryCPtr>(&output));
         dnnl::stream loc_stream(engine, dnnl::stream::flags::in_order);
-        executor->exec(loc_stream);
+        if (!executor->exec(loc_stream)) {
+            OPENVINO_THROW("Reorder failed because could not make onednn reorder!");
+        }
     }
 }