From e23c288e91d98a958fe68ea5db3e6fe40ceb482f Mon Sep 17 00:00:00 2001 From: Eli Amesefe Date: Wed, 12 Mar 2025 09:53:26 -0700 Subject: [PATCH] Add small check when input type is a list Summary: Avoids error when input is list of tensors Differential Revision: D70360605 --- backends/arm/_passes/arm_pass_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backends/arm/_passes/arm_pass_utils.py b/backends/arm/_passes/arm_pass_utils.py index a8d06713678..efb38bcd6da 100644 --- a/backends/arm/_passes/arm_pass_utils.py +++ b/backends/arm/_passes/arm_pass_utils.py @@ -149,7 +149,7 @@ def get_first_fake_tensor(node: torch.fx.Node) -> FakeTensor: If the node contains many fake tensors, return the first one. """ if isinstance( - node.meta["val"], (tuple, torch.fx.immutable_collections.immutable_list) + node.meta["val"], (list, tuple, torch.fx.immutable_collections.immutable_list) ): fake_tensor = node.meta["val"][0] else: