1 | #include <torch/csrc/profiler/api.h> |
---|---|
2 | |
3 | namespace torch { |
4 | namespace profiler { |
5 | namespace impl { |
6 | |
7 | void pushNVTXCallbacks( |
8 | const ProfilerConfig& config, |
9 | const std::unordered_set<at::RecordScope>& scopes); |
10 | |
11 | } // namespace impl |
12 | } // namespace profiler |
13 | } // namespace torch |
14 |