浏览代码

fix dcu core dumped (#4387)

zhang-prog 3 月之前
父节点
当前提交
f6294e2667
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      paddlex/inference/models/common/static_infer.py

+ 1 - 0
paddlex/inference/models/common/static_infer.py

@@ -437,6 +437,7 @@ class PaddleInfer(StaticInfer):
                 if hasattr(config, "enable_new_ir"):
                     config.enable_new_ir(self._option.enable_new_ir)
                 config.enable_use_gpu(100, self._option.device_id)
+                config.disable_mkldnn()
                 if hasattr(config, "enable_new_executor"):
                     config.enable_new_executor()
                 # XXX: is_compiled_with_rocm() must be True on dcu platform ?