Explicitly setting MKL-DNN (#15790)

This commit is contained in:
Lin Manhui 2025-06-24 22:55:56 +08:00 committed by GitHub
parent 1889d333a7
commit 86076a3746
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -89,6 +89,7 @@ def prepare_common_init_args(model_name, common_args):
elif device_type == "cpu":
enable_mkldnn = common_args["enable_mkldnn"]
if enable_mkldnn:
pp_option.run_mode = "mkldnn"
pp_option.mkldnn_cache_capacity = common_args["mkldnn_cache_capacity"]
else:
pp_option.run_mode = "paddle"