Skip to content

Commit 489eff2

Browse files
Reza Rahimifacebook-github-bot
authored andcommitted
Fix constant propagation pass to consider kwargs
Reviewed By: cccclai, mcremon-meta Differential Revision: D59345251
1 parent 5584b9e commit 489eff2

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

exir/passes/constant_prop_pass.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -126,6 +126,10 @@ def get_propagated_const_tensor_dict(
126126
node.args,
127127
exported_program,
128128
const_node_to_tensor,
129+
) or not is_const(
130+
node.kwargs,
131+
exported_program,
132+
const_node_to_tensor,
129133
):
130134
continue
131135

0 commit comments

Comments
 (0)