Преглед изворни кода

fix mkldnn (#4096)

Co-authored-by: bukejiyu <395822456@qq.com>
zhangyubo0722 пре 5 месеци
родитељ
комит
f3079d0b0b
1 измењених фајлова са 1 додато и 0 уклоњено
  1. 1 0
      paddlex/inference/models/common/static_infer.py

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

@@ -349,6 +349,7 @@ class PaddleInfer(StaticInfer):
                     if self._option.run_mode == "paddle_fp16"
                     else PrecisionType.Float32
                 )
+                config.disable_mkldnn()
                 config.enable_use_gpu(100, self._option.device_id, precision)
                 if hasattr(config, "enable_new_ir"):
                     config.enable_new_ir(self._option.enable_new_ir)