Browse Source

only split dynamic shape node for ascend

tags/v1.2.0-rc1
kswang 4 years ago
parent
commit
b6ee7f3fb2
1 changed files with 5 additions and 0 deletions
  1. +5
    -0
      mindspore/ccsrc/vm/graph_partition.cc

+ 5
- 0
mindspore/ccsrc/vm/graph_partition.cc View File

@@ -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;


Loading…
Cancel
Save