From 7f2dba9fbf88b1c36f4480e6cf1dabdb4fe7ebc4 Mon Sep 17 00:00:00 2001 From: Ethan Ng Date: Fri, 3 Oct 2025 16:30:52 -0700 Subject: [PATCH] Update ReplaceSingleElementTensorArgumentsFromFullOpWithScalarPass to check if is_tensor() is valid Differential Revision: D83861005 --- backends/cadence/aot/replace_ops.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/cadence/aot/replace_ops.py b/backends/cadence/aot/replace_ops.py index 2104764cd14..24390da5e16 100644 --- a/backends/cadence/aot/replace_ops.py +++ b/backends/cadence/aot/replace_ops.py @@ -1590,7 +1590,7 @@ def call_operator(self, op, args, kwargs, meta): updated_args = list(args) for op_arg_index in args_to_be_replaced: arg = args[op_arg_index] - if not arg.is_tensor(): + if not isinstance(arg, ProxyValue) or not arg.is_tensor(): return super().call_operator(op, args, kwargs, meta) if not isinstance(arg.node.target, EdgeOpOverload):