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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
|
#pragma once
#include <torch/csrc/WindowsTorchApiMacro.h>
#include <sstream>
#include <stdexcept>
// Forward declarations of types
namespace torch {
namespace jit {
namespace tensorexpr {
class Expr;
class Stmt;
} // namespace tensorexpr
} // namespace jit
} // namespace torch
// Forward declarations of functions
namespace std {
TORCH_API std::string to_string(const torch::jit::tensorexpr::Expr*);
TORCH_API std::string to_string(const torch::jit::tensorexpr::Stmt*);
} // namespace std
namespace torch {
namespace jit {
namespace tensorexpr {
class unsupported_dtype : public std::runtime_error {
public:
explicit unsupported_dtype() : std::runtime_error("UNSUPPORTED DTYPE") {}
explicit unsupported_dtype(const std::string& err)
: std::runtime_error("UNSUPPORTED DTYPE: " + err) {}
};
class out_of_range_index : public std::runtime_error {
public:
explicit out_of_range_index() : std::runtime_error("OUT OF RANGE INDEX") {}
explicit out_of_range_index(const std::string& err)
: std::runtime_error("OUT OF RANGE INDEX: " + err) {}
};
class unimplemented_lowering : public std::runtime_error {
public:
explicit unimplemented_lowering()
: std::runtime_error("UNIMPLEMENTED LOWERING") {}
explicit unimplemented_lowering(const Expr* expr)
: std::runtime_error("UNIMPLEMENTED LOWERING: " + std::to_string(expr)) {}
explicit unimplemented_lowering(const Stmt* stmt)
: std::runtime_error("UNIMPLEMENTED LOWERING: " + std::to_string(stmt)) {}
};
class malformed_input : public std::runtime_error {
public:
explicit malformed_input() : std::runtime_error("MALFORMED INPUT") {}
explicit malformed_input(const std::string& err)
: std::runtime_error("MALFORMED INPUT: " + err) {}
explicit malformed_input(const Expr* expr)
: std::runtime_error("MALFORMED INPUT: " + std::to_string(expr)) {}
explicit malformed_input(const std::string& err, const Expr* expr)
: std::runtime_error(
"MALFORMED INPUT: " + err + " - " + std::to_string(expr)) {}
explicit malformed_input(const Stmt* stmt)
: std::runtime_error("MALFORMED INPUT: " + std::to_string(stmt)) {}
explicit malformed_input(const std::string& err, const Stmt* stmt)
: std::runtime_error(
"MALFORMED INPUT: " + err + " - " + std::to_string(stmt)) {}
};
} // namespace tensorexpr
} // namespace jit
} // namespace torch
|
