Skip to content

Commit 6c62d31

Browse files
committed
refactor: remove unnecessary gm._code update after AST replacement
1 parent d3060c0 commit 6c62d31

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

graph_net/torch/backend/unstable_to_stable_backend.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -296,11 +296,6 @@ def visit_Call(self, node):
296296

297297
gm.forward = types.MethodType(forward_func, gm)
298298

299-
# Use serialize_graph_module_to_str to get the serialized code
300-
# This ensures the code is properly serialized with unstable API replacements
301-
serialized_code = serialize_graph_module_to_str(gm)
302-
gm._code = serialized_code
303-
304299
return gm
305300

306301
# replace this line with modification code for task 116 (torch._C._linalg.linalg_vector_norm)

0 commit comments

Comments
 (0)