Browse Source

fix dcu core dumped (#4387)

zhang-prog 3 tháng trước cách đây
mục cha
commit
f6294e2667
1 tập tin đã thay đổi với 1 bổ sung0 xóa
  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 ?