xref: /aosp_15_r20/external/pytorch/torch/csrc/itt_wrapper.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #ifndef PROFILER_ITT_H
2 #define PROFILER_ITT_H
3 #include <c10/macros/Export.h>
4 
5 namespace torch::profiler {
6 TORCH_API bool itt_is_available();
7 TORCH_API void itt_range_push(const char* msg);
8 TORCH_API void itt_range_pop();
9 TORCH_API void itt_mark(const char* msg);
10 } // namespace torch::profiler
11 
12 #endif // PROFILER_ITT_H
13