From 55b6205ddff226cea89990be6c55f28869ac33f9 Mon Sep 17 00:00:00 2001 From: Pei Yang Date: Wed, 29 Jul 2020 11:22:13 +0800 Subject: [PATCH] add set_mkldnn_cache_capacity python api(#25524) --- 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 ed6e18699d..696da67c9c 100644 --- a/paddle/fluid/pybind/inference_api.cc +++ b/paddle/fluid/pybind/inference_api.cc @@ -451,6 +451,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)