From c5501caaa38d20129bb761843644430678ce62a4 Mon Sep 17 00:00:00 2001 From: Gasoonjia Date: Tue, 16 Sep 2025 18:21:34 -0700 Subject: [PATCH] solved serde issue when serializing op using list of faketensor as output Differential Revision: D82473504 Pull Request resolved: https://github.com/pytorch/executorch/pull/14308 (cherry picked from commit ed482bd8853cd55d82761a5572362231e54d4543) --- exir/serde/export_serialize.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/exir/serde/export_serialize.py b/exir/serde/export_serialize.py index b8784cc693e..caab322d27b 100644 --- a/exir/serde/export_serialize.py +++ b/exir/serde/export_serialize.py @@ -1211,7 +1211,7 @@ def serialize_hoo_outputs(self, node: torch.fx.Node) -> List[Argument]: """ meta_val = node.meta["val"] - if isinstance(meta_val, tuple): + if isinstance(meta_val, (tuple, list)): # Note: Since we don't have a schema, we just serialize all tuple # outputs to be a list of values. Even if the output is supposed to # be a tensor list (Tensor[]), we will serialize it to be a list of