We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c997fe4 commit 54bfd72Copy full SHA for 54bfd72
backends/cadence/aot/replace_ops.py
@@ -89,10 +89,10 @@ def replace_logical_nop_where_with_where(
89
90
# Get the third arg node and its input
91
logical_not_node = node.args[0]
92
- logical_not_input_tensor = logical_not_node.args[0].to_tensor()
+ logical_not_input_node = logical_not_node.args[0]
93
94
# If the logical_not input is not a boolean tensor, bail.
95
- if logical_not_input_tensor.meta["spec"].dtype != torch.bool:
+ if logical_not_input_node.meta["val"].dtype != torch.bool:
96
continue
97
98
# Replace the where op with another one, flipping the inputs and using the boolean
0 commit comments