From 0722297d9b8c3df4407ce8de40adbdc71ffdce4e Mon Sep 17 00:00:00 2001 From: Wilber Date: Wed, 23 Jun 2021 10:34:52 +0800 Subject: [PATCH] modify mkldnn default capacity (#33729) --- paddle/fluid/inference/api/paddle_analysis_config.h | 2 +- paddle/fluid/inference/goapi/go.mod | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/inference/api/paddle_analysis_config.h b/paddle/fluid/inference/api/paddle_analysis_config.h index ae29b4ff64c..81e742e8a6f 100644 --- a/paddle/fluid/inference/api/paddle_analysis_config.h +++ b/paddle/fluid/inference/api/paddle_analysis_config.h @@ -699,7 +699,7 @@ struct PD_INFER_DECL AnalysisConfig { bool xpu_adaptive_seqlen_; // mkldnn related. - int mkldnn_cache_capacity_{0}; + int mkldnn_cache_capacity_{10}; bool use_mkldnn_quantizer_{false}; std::shared_ptr mkldnn_quantizer_config_; bool use_mkldnn_bfloat16_{false}; diff --git a/paddle/fluid/inference/goapi/go.mod b/paddle/fluid/inference/goapi/go.mod index 1036a2e3281..96e04486f0f 100644 --- a/paddle/fluid/inference/goapi/go.mod +++ b/paddle/fluid/inference/goapi/go.mod @@ -1,3 +1,3 @@ -module github.com/jiweibo/paddle/paddle/fluid/inference/goapi +module github.com/paddlepaddle/paddle/paddle/fluid/inference/goapi go 1.15 -- GitLab