Skip to content

Commit a5f64cc

Browse files
author
ssjia
committed
[etrecord] Implement generic fallback for GraphModuleSerializer.handle_call_function
Title says it all! Implement the case where `node.target` is neither `torch._ops.OpOverload` or `torch._ops.HigherOrderOperator`, instead of throwing an exception. Differential Revision: [D88216198](https://our.internmc.facebook.com/intern/diff/D88216198/) [ghstack-poisoned]
1 parent 33ec615 commit a5f64cc

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

exir/serde/export_serialize.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -546,7 +546,13 @@ def handle_call_function(self, node: torch.fx.Node):
546546
metadata=self.serialize_metadata(node),
547547
)
548548
else:
549-
raise SerializeError(f"Serializing {node.target} is not supported")
549+
ex_node = Node(
550+
name=node.name,
551+
target=node._pretty_print_target(node.target),
552+
inputs=self.serialize_hoo_inputs(node.args, node.kwargs),
553+
outputs=self.serialize_hoo_outputs(node),
554+
metadata=self.serialize_metadata(node),
555+
)
550556

551557
self.graph_state.nodes.append(ex_node)
552558

0 commit comments

Comments
 (0)