Skip to content

Commit ac5dd30

Browse files
Olivia-liufacebook-github-bot
authored andcommitted
Fix exir deserialization logic (#2857)
Summary: Pull Request resolved: #2857 Looks like a recent str -> TensorArgument change in diff D54855251 wasn't very thorough. ETRecord deserialization failed because of it. Reviewed By: Jack-Khuu, angelayi Differential Revision: D55776877 fbshipit-source-id: 2e0428a1c0cd0e7e2e8d7a838ba611020b5716a1
1 parent 033f962 commit ac5dd30

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
@@ -1503,7 +1503,7 @@ def deserialize_optional_tensor_args(a):
15031503
if a.type == "as_none":
15041504
return None
15051505
elif a.type == "as_tensor":
1506-
return self.serialized_name_to_node[a.value]
1506+
return self.serialized_name_to_node[a.value.name]
15071507
else:
15081508
raise SerializeError(f"Unhandled argument {inp}")
15091509

0 commit comments

Comments
 (0)