16 lines
467 B
C++
16 lines
467 B
C++
#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
|