[SOT] Avoid deep copy while creating MutableDictLikeData for DataClassInstanceVariable (#73407)

0x3878f · web-flow · commit 4ad3b8f87edc · 2025-06-18T11:40:39.000+08:00
diff --git a/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py b/python/paddle/jit/sot/opcode_translator/executor/variables/basic.py
@@ -18,7 +18,7 @@
 import operator
 import sys
 import types
-from dataclasses import asdict, is_dataclass
+from dataclasses import is_dataclass
 from enum import Enum
 from functools import cached_property, reduce
 from typing import TYPE_CHECKING, Any
@@ -2563,14 +2563,18 @@ def flatten_inner_vars(self) -> list[VariableBase]:
             for fd in dataclasses.fields(self.get_py_type())
         ]
 
+    @classmethod
+    def _custom_asdict(cls, obj):
+        return {f.name: getattr(obj, f.name) for f in dataclasses.fields(obj)}
+
     @VariableFactory.register_from_value()
     def from_value(value: object, graph: FunctionGraph, tracker: Tracker):
         if is_dataclass(value) and not isinstance(value, type):
             class_var = VariableFactory.from_value(
                 type(value), graph, DanglingTracker()
             )
             var = DataClassInstanceVariable(
-                asdict(value),
+                DataClassInstanceVariable._custom_asdict(value),
                 class_var,
                 id(value),
                 graph=graph,
