blob: 09d09074b726d9928207d32ba98f2d4439f99b5e [file] [log] [blame]
#include <ATen/native/verbose_wrapper.h>
#include <torch/csrc/utils/pybind.h>
#include <torch/csrc/utils/verbose.h>
namespace torch {
void initVerboseBindings(PyObject* module) {
auto m = py::handle(module).cast<py::module>();
auto verbose = m.def_submodule("_verbose", "MKL, MKLDNN verbose");
verbose.def("mkl_set_verbose", torch::verbose::_mkl_set_verbose);
verbose.def("mkldnn_set_verbose", torch::verbose::_mkldnn_set_verbose);
}
} // namespace torch