Skip to content

Commit ed482bd

Browse files
authored
solved serde issue when serializing op using list of faketensor as output
Differential Revision: D82473504 Pull Request resolved: #14308
1 parent 9c12e69 commit ed482bd

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)