add set_mkldnn_cache_capacity python api(#25524)

fix_copy_if_different
Pei Yang 5 years ago committed by GitHub
parent e0a9115e28
commit 55b6205ddf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -451,6 +451,8 @@ void BindAnalysisConfig(py::module *m) {
#ifdef PADDLE_WITH_MKLDNN #ifdef PADDLE_WITH_MKLDNN
.def("quantizer_config", &AnalysisConfig::mkldnn_quantizer_config, .def("quantizer_config", &AnalysisConfig::mkldnn_quantizer_config,
py::return_value_policy::reference) py::return_value_policy::reference)
.def("set_mkldnn_cache_capacity", &AnalysisConfig::SetMkldnnCacheCapacity,
py::arg("capacity") = 0)
#endif #endif
.def("set_mkldnn_op", &AnalysisConfig::SetMKLDNNOp) .def("set_mkldnn_op", &AnalysisConfig::SetMKLDNNOp)
.def("set_model_buffer", &AnalysisConfig::SetModelBuffer) .def("set_model_buffer", &AnalysisConfig::SetModelBuffer)

Loading…
Cancel
Save