1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
#include <c10/util/irange.h>
#include <torch/csrc/autograd/profiler.h>
#include <torch/csrc/itt_wrapper.h>
#include <sstream>
namespace torch {
namespace profiler {
namespace impl {
namespace {
struct ITTMethods : public ProfilerStubs {
void record(int* device, ProfilerEventStub* event, int64_t* cpu_ns)
const override {}
float elapsed(const ProfilerEventStub* event, const ProfilerEventStub* event2)
const override {
return 0;
}
void mark(const char* name) const override {
// NOLINTNEXTLINE(cppcoreguidelines-init-variables)
torch::profiler::itt_mark(name);
}
void rangePush(const char* name) const override {
// NOLINTNEXTLINE(cppcoreguidelines-init-variables)
torch::profiler::itt_range_push(name);
}
void rangePop() const override {
torch::profiler::itt_range_pop();
}
void onEachDevice(std::function<void(int)> op) const override {}
void synchronize() const override {}
bool enabled() const override {
return true;
}
};
struct RegisterITTMethods {
RegisterITTMethods() {
static ITTMethods methods;
registerITTMethods(&methods);
}
};
RegisterITTMethods reg;
} // namespace
} // namespace impl
} // namespace profiler
} // namespace torch
|
