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