Skip to content

Commit 92dee40

Browse files
fix mkldnn (#4098)
Co-authored-by: bukejiyu <[email protected]>
1 parent 53456fc commit 92dee40

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

paddlex/inference/models/common/static_infer.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -349,6 +349,7 @@ def _create(
349349
if self._option.run_mode == "paddle_fp16"
350350
else PrecisionType.Float32
351351
)
352+
config.disable_mkldnn()
352353
config.enable_use_gpu(100, self._option.device_id, precision)
353354
if hasattr(config, "enable_new_ir"):
354355
config.enable_new_ir(self._option.enable_new_ir)

0 commit comments

Comments
 (0)