Merge pull request #15 from 0x45f/build-strategy

Support build_strategy for sot
feifei-111 · Jun 27, 2023 · dfef6dc · dfef6dc
2 parents 762547c + ca763f3
commit dfef6dc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/paddle/jit/dy2static/program_translator.py b/python/paddle/jit/dy2static/program_translator.py
@@ -678,7 +678,10 @@ def __init__(self, function, input_spec=None, **kwargs):
     def _perform_call(self, *args, **kwargs):
         from sot import symbolic_translate
 
-        traced_fun = symbolic_translate(self._dygraph_function)
+        build_strategy = self._kwargs.get("build_strategy", None)
+        traced_fun = symbolic_translate(
+            self._dygraph_function, build_strategy=build_strategy
+        )
         if self._class_instance is not None:
             args = (self._class_instance,) + args
         return traced_fun(*args, **kwargs)