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

fix elemany bug

This commit is contained in:
Yang Jiao 2022-05-17 17:06:39 +08:00
parent e10ebf758e
commit 9fd93571b1
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

@ -1001,8 +1001,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