Skip to content

Commit e507bf9

Browse files
solved serde issue when serializing op using list of faketensor as output (#14366)
Summary: as title. Differential Revision: D82473504 Co-authored-by: Gasoonjia <[email protected]>
1 parent b752a77 commit e507bf9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

exir/serde/export_serialize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1211,7 +1211,7 @@ def serialize_hoo_outputs(self, node: torch.fx.Node) -> List[Argument]:
12111211
"""
12121212
meta_val = node.meta["val"]
12131213

1214-
if isinstance(meta_val, tuple):
1214+
if isinstance(meta_val, (tuple, list)):
12151215
# Note: Since we don't have a schema, we just serialize all tuple
12161216
# outputs to be a list of values. Even if the output is supposed to
12171217
# be a tensor list (Tensor[]), we will serialize it to be a list of

0 commit comments

Comments
 (0)