From 54100355b2a43593e75a41a468df6f98a420f2f1 Mon Sep 17 00:00:00 2001
From: Pei Yang <peiyang@baidu.com>
Date: Wed, 29 Jul 2020 11:38:00 +0800
Subject: [PATCH] add set_mkldnn_cache_capacity python api (#25772)

---
 paddle/fluid/pybind/inference_api.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/paddle/fluid/pybind/inference_api.cc b/paddle/fluid/pybind/inference_api.cc
index 2a1de8c672..e4927977aa 100644
--- a/paddle/fluid/pybind/inference_api.cc
+++ b/paddle/fluid/pybind/inference_api.cc
@@ -442,6 +442,8 @@ void BindAnalysisConfig(py::module *m) {
 #ifdef PADDLE_WITH_MKLDNN
       .def("quantizer_config", &AnalysisConfig::mkldnn_quantizer_config,
            py::return_value_policy::reference)
+      .def("set_mkldnn_cache_capacity", &AnalysisConfig::SetMkldnnCacheCapacity,
+           py::arg("capacity") = 0)
 #endif
       .def("set_mkldnn_op", &AnalysisConfig::SetMKLDNNOp)
       .def("set_model_buffer", &AnalysisConfig::SetModelBuffer)
-- 
GitLab