Skip to content

Commit 69de566

Browse files
committed
update instead of union
Signed-off-by: Justin Chu <[email protected]>
1 parent 55c47d9 commit 69de566

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

onnxscript/optimizer/_constant_folding.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -922,7 +922,7 @@ def _record_contributing_values(original_node: ir.Node, replacement: Replacement
922922
for input in original_node.inputs:
923923
if input is None:
924924
continue
925-
folded_from.union(input.meta.get(_FOLDED_FROM_KEY, set()))
925+
folded_from.update(input.meta.get(_FOLDED_FROM_KEY, set()))
926926
assert input.name is not None
927927
folded_from.add(input.name)
928928

0 commit comments

Comments
 (0)