From 0e2a4d5757c175c023e58a7a8992aac0587479d5 Mon Sep 17 00:00:00 2001 From: WangZhen <23097963+0x45f@users.noreply.github.com> Date: Thu, 10 Aug 2023 14:15:39 +0800 Subject: [PATCH] Pass backend arg to symbolic_translate (#56107) --- python/paddle/jit/dy2static/program_translator.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/paddle/jit/dy2static/program_translator.py b/python/paddle/jit/dy2static/program_translator.py index 113daf4cb1b..6ecdc8cb26a 100644 --- a/python/paddle/jit/dy2static/program_translator.py +++ b/python/paddle/jit/dy2static/program_translator.py @@ -709,8 +709,11 @@ class SymbolicStaticFunction(StaticFunction): from sot import symbolic_translate build_strategy = self._kwargs.get("build_strategy", None) + backend = self._kwargs.get("backend", None) traced_fun = symbolic_translate( - self._dygraph_function, build_strategy=build_strategy + self._dygraph_function, + build_strategy=build_strategy, + backend=backend, ) if self._class_instance is not None: args = (self._class_instance,) + args -- GitLab