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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
#include <c10/core/SymBool.h>
#include <c10/core/SymNodeImpl.h>
namespace c10 {
SymNode SymBool::toSymNodeImpl() const {
TORCH_CHECK(is_heap_allocated());
return SymNode::reclaim_copy(toSymNodeImplUnowned());
}
SymNode SymBool::wrap_node(const SymNode& base) const {
if (auto ma = maybe_as_bool()) {
return base->wrap_bool(*ma);
} else {
return toSymNodeImpl();
}
}
#define DEFINE_BINARY(API, OP, METHOD, RET) \
RET SymBool::API(const SymBool& sci) const { \
if (auto ma = maybe_as_bool()) { \
if (auto mb = sci.maybe_as_bool()) { \
return RET(OP(*ma, *mb)); \
} else { \
auto b = sci.toSymNodeImpl(); \
return RET(b->wrap_bool(*ma)->METHOD(b)); \
} \
} else { \
if (auto mb = sci.maybe_as_bool()) { \
auto a = toSymNodeImplUnowned(); \
return RET(a->METHOD(a->wrap_bool(*mb))); \
} else { \
return RET(toSymNodeImplUnowned()->METHOD(sci.toSymNodeImpl())); \
} \
} \
}
// clang-format off
DEFINE_BINARY(sym_and, std::logical_and<>(), sym_and, SymBool)
DEFINE_BINARY(sym_or, std::logical_or<>(), sym_or, SymBool)
// clang-format on
SymBool SymBool::sym_not() const {
if (auto ma = maybe_as_bool()) {
return SymBool(!*ma);
}
return SymBool(toSymNodeImpl()->sym_not());
}
std::ostream& operator<<(std::ostream& os, const SymBool& s) {
if (auto ma = s.maybe_as_bool()) {
os << *ma;
} else {
os << s.toSymNodeImpl()->str();
}
return os;
}
bool SymBool::guard_bool(const char* file, int64_t line) const {
if (auto ma = maybe_as_bool()) {
return *ma;
}
SymNode a = toSymNodeImpl();
return a->guard_bool(file, line);
}
bool SymBool::guard_size_oblivious(const char* file, int64_t line) const {
if (auto ma = maybe_as_bool()) {
return *ma;
}
SymNode a = toSymNodeImpl();
return a->guard_size_oblivious(file, line);
}
bool SymBool::expect_true(const char* file, int64_t line) const {
if (auto ma = maybe_as_bool()) {
return *ma;
}
SymNode a = toSymNodeImpl();
return a->expect_true(file, line);
}
bool SymBool::has_hint() const {
if (maybe_as_bool()) {
return true;
}
return toSymNodeImpl()->has_hint();
}
} // namespace c10
|
