From b48738297541d9e32924768d076f325285f722f9 Mon Sep 17 00:00:00 2001 From: Songhao Jia Date: Mon, 15 Sep 2025 12:32:12 -0700 Subject: [PATCH] solved serde issue when serializing op using list of faketensor as output Summary: as title. Differential Revision: D82473504 --- 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