saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[exir] Initial serialization (#103763)

Browse source 
Summary:
ETRecord can't use this yet because the other programs need to be migrated to using ExportedProgram (D46729844)

Note: higher order ops like call_delegate/cond are also not supported yet

Test Plan: `buck2 run @//mode/dev-nosan //executorch/exir/tests:serde`

Differential Revision: D46802454

Pull Request resolved: https://github.com/pytorch/pytorch/pull/103763
Approved by: https://github.com/tarun292
This commit is contained in:
Angela Yi 2023-06-26 18:05:27 +00:00 committed by PyTorch MergeBot
parent 618cc82e77
commit dd4f4bb47d
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torch/_export/serde/serialize.py
View file

@ -450,7 +450,7 @@ class GraphModuleSerializer:
node_args = node.args[0]
if isinstance(node_args, torch.fx.Node):
node_args = (node_args,)
assert isinstance(node_args, tuple)
assert isinstance(node_args, (tuple, list))
self.graph_state.outputs = [self.serialize_input(arg) for arg in node_args]
def handle_call_function(self, node: torch.fx.Node):