[XPU] Open ir pass (#8598)

PaddlePaddle · Jun 13, 2024 · 2dc4d7b · 2dc4d7b
1 parent 8e24341
commit 2dc4d7b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/llm/predictor.py b/llm/predictor.py
@@ -1092,6 +1092,8 @@ def _create_predictor(self, predictor_args: PredictorArgument):
             xpu_config.l3_size = 63 * 1024 * 1024
             xpu_config.l3_autotune_size = 63 * 1024 * 1024
             config.set_xpu_config(xpu_config)
+            config.switch_ir_optim(True)
+            config.enable_memory_optim()
         else:
             device_id = int(os.environ.get("FLAGS_selected_gpus", 0))
             config.enable_use_gpu(100, device_id)