瀏覽代碼

support to enable new_ir on CPU device

gaotingquan 1 年之前
父節點
當前提交
11d2e39f4a
共有 1 個文件被更改,包括 2 次插入0 次删除
  1. 2 0
      paddlex/modules/base/predictor/utils/paddle_inference_predictor.py

+ 2 - 0
paddlex/modules/base/predictor/utils/paddle_inference_predictor.py

@@ -68,6 +68,8 @@ class _PaddleInferencePredictor(object):
         else:
             assert option.device == "cpu"
             config.disable_gpu()
+            config.enable_new_ir(True)
+            config.enable_new_executor(True)
             if "mkldnn" in option.run_mode:
                 try:
                     config.enable_mkldnn()