add register_from_value

DrRyanHuang · DrRyanHuang · commit 39ea9870882f · 2025-05-29T08:30:49.000Z
diff --git a/python/paddle/jit/sot/opcode_translator/executor/variables/callable.py b/python/paddle/jit/sot/opcode_translator/executor/variables/callable.py
@@ -1263,6 +1263,7 @@ def call_function(self, /, *args, **kwargs):
             obj, self.graph, CreateLayerTracker(self, args, kwargs)
         )
 
+    @VariableFactory.register_from_value()
     def from_value(value: partial, graph: FunctionGraph, tracker: Tracker):
         if isinstance(value, partial):
             return PartialVariable(value, graph, tracker)

Original file line number	Diff line number	Diff line change
`@@ -1263,6 +1263,7 @@ def call_function(self, /, args, *kwargs):`
`1263`	`1263`	`obj, self.graph, CreateLayerTracker(self, args, kwargs)`
`1264`	`1264`	`)`
`1265`	`1265`
	`1266`	`+ @VariableFactory.register_from_value()`
`1266`	`1267`	`def from_value(value: partial, graph: FunctionGraph, tracker: Tracker):`
`1267`	`1268`	`if isinstance(value, partial):`
`1268`	`1269`	`return PartialVariable(value, graph, tracker)`