提交 8452d175 authored 作者: Frederic Bastien's avatar Frederic Bastien 提交者: Reyhane Askari

stop the comparison earlier

上级 29416c10
...@@ -652,6 +652,8 @@ class MergeFeature(object): ...@@ -652,6 +652,8 @@ class MergeFeature(object):
continue continue
if len(node.inputs) != len(candidate.inputs): if len(node.inputs) != len(candidate.inputs):
continue continue
if node.op != candidate.op:
continue
cand_has_assert = False cand_has_assert = False
...@@ -688,7 +690,7 @@ class MergeFeature(object): ...@@ -688,7 +690,7 @@ class MergeFeature(object):
in zip(node_inputs_assert_removed, in zip(node_inputs_assert_removed,
cand_inputs_assert_removed)) cand_inputs_assert_removed))
if inputs_match and node.op == candidate.op: if inputs_match:
if (node, candidate) in self.blacklist: if (node, candidate) in self.blacklist:
# They were already tried, and there was an error # They were already tried, and there was an error
continue continue
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论