fix

PaddlePaddle · Apr 29, 2024 · 38d9652 · 38d9652
1 parent 24979ea
commit 38d9652
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/paddle/pir_utils.py b/python/paddle/pir_utils.py
@@ -73,6 +73,7 @@ def __enter__(self):
             paddle.enable_static()
         if not self.old_flag:
             paddle.framework.set_flags({"FLAGS_enable_pir_api": True})
+            paddle.framework.set_flags({"FLAGS_enable_pir_in_executor": True})
             paddle.base.framework.global_var._use_pir_api_ = True
             bind_datatype()
             self._switch_to_pir()