Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!34515 Fix Floatstatus Bug 

Merge pull request !34515 from jiaoy1224/real_master
This commit is contained in:
i-robot 2022-05-18 06:29:07 +00:00 committed by Gitee
parent 49317d8022 9fd93571b1
commit efd0972b66
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/python/mindspore/_extends/graph_kernel/model/graph_split.py
View File

@ -981,8 +981,8 @@ class GraphSplitGpu(GraphSplitByPattern):
return None
fused = []
for a, _ in dom.in_relations.items():
if a.pattern < PrimLib.BROADCAST and a.check_acyclic(dom):
for a, r in dom.in_relations.items():
if a.pattern < PrimLib.BROADCAST and r <= PrimLib.ELEMWISE and a.check_acyclic(dom):
fused.append(a)
return fused, True