From 735642f1d5f34bf497334c79f3d2d8ec463e6ad0 Mon Sep 17 00:00:00 2001
From: hwjiaorui <jiaorui3@huawei.com>
Date: Sat, 20 Nov 2021 14:22:07 +0800
Subject: [PATCH] add dynamic shape check

---
 mindspore/ccsrc/backend/session/ascend_session.cc | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/mindspore/ccsrc/backend/session/ascend_session.cc b/mindspore/ccsrc/backend/session/ascend_session.cc
index 692f65cb903..47d0bf18d69 100644
--- a/mindspore/ccsrc/backend/session/ascend_session.cc
+++ b/mindspore/ccsrc/backend/session/ascend_session.cc
@@ -279,6 +279,17 @@ void AddGraphToManager(const NotNull<KernelGraphPtr> graph, NotNull<FuncGraphMan
     AddGraphToManager(NOT_NULL(child_graph.lock()), manager, memo);
   }
 }
+void CheckControlFlowDynamicShape(std::vector<KernelGraphPtr> all_graphs) {
+  if (all_graphs.size() <= 1) {
+    return;
+  }
+  for (auto &graph : all_graphs) {
+    if (graph->is_dynamic_shape()) {
+      MS_LOG(EXCEPTION) << "Dynamic shape is not supported with control flow(loop control statements and conditions "
+                           "control statements).";
+    }
+  }
+}
 }  // namespace
 
 void AscendSession::Init(uint32_t device_id) { InitExecutor(kAscendDevice, device_id); }
@@ -387,6 +398,8 @@ GraphId AscendSession::CompileGraphImpl(NotNull<FuncGraphPtr> func_graph) {
   }
   UnifyMindIR(root_graph);
   opt::BackendCommonOptimization(root_graph);
+  CheckControlFlowDynamicShape(all_graphs);
+
   // empty graph dont entry to backend
   if (root_graph->execution_order().empty()) {
     MS_LOG(INFO) << root_graph->ToString() << " is empty graph.";