set

pytorch · Jul 27, 2023 · 9ad1f76 · 9ad1f76
1 parent 9b32140
commit 9ad1f76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch/_export/serde/upgrade.py b/torch/_export/serde/upgrade.py
@@ -197,6 +197,6 @@ def upgrade(self, exported_program: ep.ExportedProgram) -> ep.ExportedProgram:
             # NB: we have to retrace the graph_module instead of ep because of some failure. Also, we need to turn of
             # _add_runtime_assertions because dynamo is not happy with sym_size.int.
             exported_program = export(upgraded_program.graph_module, inputs, {})
-            exported_program.call_spec = upgraded_program.call_spec
+            exported_program._call_spec = upgraded_program.call_spec
 
         return exported_program