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
|
#pragma once
#include <c10/macros/Macros.h>
#include <string>
#include <vector>
namespace torch::lazy {
struct SourceLocation {
std::string file;
std::string function;
int line = -1;
};
TORCH_API void EmitShortFrameInfo(
std::ostream& stream,
const std::vector<SourceLocation>& frames);
TORCH_API std::ostream& operator<<(
std::ostream& stream,
const std::vector<SourceLocation>& frames);
// The base class for user defined metadata which is possible to attach to IR
// nodes.
struct TORCH_API UserMetaData {
virtual ~UserMetaData() = default;
};
struct TORCH_API MetaData {
std::string scope;
std::vector<SourceLocation> frame_info;
};
// TODO(whc) is this going to be used outside of in IR decompositions?
// RAII data structure to be used a stack variable to enter a new IR scope. IR
// scope names will appear in the IR and will help identifying the source of the
// single IR nodes.
struct TORCH_API ScopePusher {
explicit ScopePusher(const std::string& name);
~ScopePusher();
ScopePusher(ScopePusher&& other) = delete;
ScopePusher(const ScopePusher&) = delete;
ScopePusher& operator=(const ScopePusher&) = delete;
ScopePusher& operator=(ScopePusher&&) = delete;
static void ResetScopes();
};
TORCH_API MetaData GetMetaDataIfDebugging();
} // namespace torch::lazy
|
