1 #include <torch/csrc/itt_wrapper.h> 2 #include <torch/csrc/utils/pybind.h> 3 4 namespace torch::profiler { initIttBindings(PyObject * module)5void initIttBindings(PyObject* module) { 6 auto m = py::handle(module).cast<py::module>(); 7 8 auto itt = m.def_submodule("_itt", "VTune ITT bindings"); 9 itt.def("is_available", itt_is_available); 10 itt.def("rangePush", itt_range_push); 11 itt.def("rangePop", itt_range_pop); 12 itt.def("mark", itt_mark); 13 } 14 } // namespace torch::profiler 15