소스 검색

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 ?