From 9fd93571b166ff81f3d8c01d9d9f114f3a756835 Mon Sep 17 00:00:00 2001 From: Yang Jiao Date: Tue, 17 May 2022 17:06:39 +0800 Subject: [PATCH] fix elemany bug --- .../mindspore/_extends/graph_kernel/model/graph_split.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/python/mindspore/_extends/graph_kernel/model/graph_split.py b/mindspore/python/mindspore/_extends/graph_kernel/model/graph_split.py index 910761bbc1e..94df7468b1b 100644 --- a/mindspore/python/mindspore/_extends/graph_kernel/model/graph_split.py +++ b/mindspore/python/mindspore/_extends/graph_kernel/model/graph_split.py @@ -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