1 | #include <torch/csrc/itt_wrapper.h> |
---|---|
2 | #include <torch/csrc/utils/pybind.h> |
3 | |
4 | namespace torch { |
5 | namespace profiler { |
6 | void 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 |