From 2971522f1b2805a59a4cfa8597a657227442558c Mon Sep 17 00:00:00 2001
From: wxl <wanxuelei@huawei.com>
Date: Mon, 8 Feb 2021 15:19:57 +0800
Subject: [PATCH] reformat need to delete

---
 ge/graph/passes/reshape_remove_pass.cc | 34 +++++++++++++++++++-------
 1 file changed, 25 insertions(+), 9 deletions(-)

diff --git a/ge/graph/passes/reshape_remove_pass.cc b/ge/graph/passes/reshape_remove_pass.cc
index ffa6a485..3268041b 100755
--- a/ge/graph/passes/reshape_remove_pass.cc
+++ b/ge/graph/passes/reshape_remove_pass.cc
@@ -15,29 +15,45 @@
  */
 
 #include "graph/passes/reshape_remove_pass.h"
+
+#include  <map>
+#include  <string>
+
 #include "framework/common/util.h"
+#include "framework/common/types.h"
 #include "graph/passes/pass_utils.h"
 #include "graph/utils/node_utils.h"
 
 namespace ge {
 namespace {
 const int kReshapeDataIndex = 0;
+const int kReshapeType = 0;
+const int kReformatType = 1;
+std::map<const char *, int> kOpTypeHash = {
+  {RESHAPE, kReshapeType},
+  {REFORMAT, kReformatType}
+};
 }
 
 Status ReshapeRemovePass::Run(NodePtr &node) {
   GE_CHECK_NOTNULL(node);
   GE_CHECK_NOTNULL(node->GetOpDesc());
-  if (node->GetType() != RESHAPE && node->GetType() != REFORMAT) {
-    return SUCCESS;
-  }
 
-  bool is_shape_unknown = false;
-  if (NodeUtils::GetNodeUnknownShapeStatus(*node, is_shape_unknown) == GRAPH_SUCCESS) {
-    if (is_shape_unknown) {
-      GELOGI("op:%s is unknown shape, can not be deleted.",
-             node->GetName().c_str());
+  switch(kOpTypeHash.find(node->GetType())) {
+    case kReshapeType:
+      bool is_shape_unknown = false;
+      if (NodeUtils::GetNodeUnknownShapeStatus(*node, is_shape_unknown) == GRAPH_SUCCESS) {
+        if (is_shape_unknown) {
+          GELOGI("op:%s is unknown shape, can not be deleted.",
+                 node->GetName().c_str());
+          return SUCCESS;
+        }
+      }
+      break;
+    case kReformatType:
+      break;
+    default:
       return SUCCESS;
-    }
   }
 
   GELOGI("Remove %s node %s", node->GetType().c_str(), node->GetName().c_str());