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 <torch/csrc/jit/frontend/tree.h>
namespace torch::jit {
struct Call {
std::string fn_name;
SourceRange caller_range;
};
struct TORCH_API ErrorReport : public std::exception {
ErrorReport(const ErrorReport& e);
explicit ErrorReport(const SourceRange& r);
explicit ErrorReport(const TreeRef& tree) : ErrorReport(tree->range()) {}
explicit ErrorReport(const Token& tok) : ErrorReport(tok.range) {}
const char* what() const noexcept override;
struct TORCH_API CallStack {
// These functions are used to report why a function was being compiled
// (i.e. what was the call stack of user functions at compilation time that
// led to this error)
CallStack(const std::string& name, const SourceRange& range);
~CallStack();
// Change the range that is relevant for the current function (i.e. after
// each successful expression compilation, change it to the next expression)
static void update_pending_range(const SourceRange& range);
};
static std::string current_call_stack();
private:
template <typename T>
friend const ErrorReport& operator<<(const ErrorReport& e, const T& t);
mutable std::stringstream ss;
OwnedSourceRange context;
mutable std::string the_message;
std::vector<Call> error_stack;
};
template <typename T>
const ErrorReport& operator<<(const ErrorReport& e, const T& t) {
e.ss << t;
return e;
}
} // namespace torch::jit
|
