mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Fixes https://github.com/pytorch/pytorch/pull/84848#discussion_r986329680 @malfet @slgong-fb Pull Request resolved: https://github.com/pytorch/pytorch/pull/86199 Approved by: https://github.com/malfet
14 lines
274 B
C++
14 lines
274 B
C++
#ifndef PROFILER_ITT_H
|
|
#define PROFILER_ITT_H
|
|
|
|
namespace torch {
|
|
namespace profiler {
|
|
bool itt_is_available();
|
|
void itt_range_push(const char* msg);
|
|
void itt_range_pop();
|
|
void itt_mark(const char* msg);
|
|
} // namespace profiler
|
|
} // namespace torch
|
|
|
|
#endif // PROFILER_ITT_H
|