1 #include <ATen/native/verbose_wrapper.h> 2 #include <torch/csrc/utils/pybind.h> 3 #include <torch/csrc/utils/verbose.h> 4 5 namespace torch { 6 initVerboseBindings(PyObject * module)7void initVerboseBindings(PyObject* module) { 8 auto m = py::handle(module).cast<py::module>(); 9 10 auto verbose = m.def_submodule("_verbose", "MKL, MKLDNN verbose"); 11 verbose.def("mkl_set_verbose", torch::verbose::_mkl_set_verbose); 12 verbose.def("mkldnn_set_verbose", torch::verbose::_mkldnn_set_verbose); 13 } 14 15 } // namespace torch 16