From b6ee7f3fb231c910b19008504e4dd81b5cd83805 Mon Sep 17 00:00:00 2001
From: kswang <wangkaisheng2@huawei.com>
Date: Thu, 7 Jan 2021 21:25:27 +0800
Subject: [PATCH] only split dynamic shape node for ascend

---
 mindspore/ccsrc/vm/graph_partition.cc | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/mindspore/ccsrc/vm/graph_partition.cc b/mindspore/ccsrc/vm/graph_partition.cc
index 51a975387f..932999816a 100644
--- a/mindspore/ccsrc/vm/graph_partition.cc
+++ b/mindspore/ccsrc/vm/graph_partition.cc
@@ -659,6 +659,11 @@ void NodesToSegments(const std::vector<AnfNodePtr> &segment_nodes, std::vector<G
   if (segment_nodes.size() == 0) {
     return;
   }
+  auto segment_target = GetCNodeTarget(segment_nodes[0]);
+  if (segment_target != kAscendDevice) {
+    AddSegment(segment_nodes, segments, node_to_segment);
+    return;
+  }
   MS_EXCEPTION_IF_NULL(segments);
   MS_EXCEPTION_IF_NULL(node_to_segment);
   std::set<AnfNodePtr> dynamic_nodes_set;