update

zhangmeng

2020-01-16 a47fccb11fa3470901aebcb27f861d242d0925e1

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
#pragma once
 
#include <c10/util/Optional.h>
#include <torch/csrc/jit/script/tree.h>
 
namespace torch {
namespace jit {
namespace script {
 
struct Call {
  std::string fn_name;
  c10::optional<SourceRange> caller_range;
};
 
struct CAFFE2_API ErrorReport : public std::exception {
  ErrorReport(const ErrorReport& e);
 
  ErrorReport();
  explicit ErrorReport(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 CAFFE2_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);
    ~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);
  };
 
 private:
  template <typename T>
  friend const ErrorReport& operator<<(const ErrorReport& e, const T& t);
 
  mutable std::stringstream ss;
  c10::optional<SourceRange> 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 script
} // namespace jit
} // namespace torch