Browse Source

!21621 fix recompute_fuse

Merge pull request !21621 from lingyunli63/fix_recompute_fuse
tags/v1.5.0-rc1
i-robot Gitee 4 years ago
parent
commit
a5388a5227
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/_extends/graph_kernel/model/graph_split.py

+ 1
- 1
mindspore/_extends/graph_kernel/model/graph_split.py View File

@@ -486,7 +486,7 @@ class GraphSplitByPattern:
cheap_regions = []
for output in outputs:
# tensor should have user other than user_area to be fused
if output.para_type != Tensor.PARA_OUTPUT and len(output.to_ops) < 2:
if len(output.to_ops) < 2:
continue
region_ops = []
grow = True


Loading…
Cancel
Save