only split dynamic shape node for ascend

4 years ago · b6ee7f3fb2
parent 62888f23b8
commit b6ee7f3fb2
1 changed files with 5 additions and 0 deletions
--- 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;