فهرست منبع

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()