Skip to content

Commit 54bfd72

Browse files
authored
Fix Wav2Vec Replace Pass Bug
Differential Revision: D83778606 Pull Request resolved: #14757
1 parent c997fe4 commit 54bfd72

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

backends/cadence/aot/replace_ops.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -89,10 +89,10 @@ def replace_logical_nop_where_with_where(
8989

9090
# Get the third arg node and its input
9191
logical_not_node = node.args[0]
92-
logical_not_input_tensor = logical_not_node.args[0].to_tensor()
92+
logical_not_input_node = logical_not_node.args[0]
9393

9494
# If the logical_not input is not a boolean tensor, bail.
95-
if logical_not_input_tensor.meta["spec"].dtype != torch.bool:
95+
if logical_not_input_node.meta["val"].dtype != torch.bool:
9696
continue
9797

9898
# Replace the where op with another one, flipping the inputs and using the boolean

0 commit comments

Comments
 (0)