forked from pytorch/pytorch
-
Notifications
You must be signed in to change notification settings - Fork 0
/
register_symbols.cpp
38 lines (30 loc) · 1.01 KB
/
register_symbols.cpp
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
#include <ATen/core/interned_strings_class.h>
namespace c10 {
namespace {
struct Entry {
const char* const qual_name;
const char* const unqual_name;
const Symbol sym;
const Symbol ns_sym;
};
constexpr Entry entries[] = {
#define SYMBOL_ENTRY(n, s) {#n "::" #s, #s, n::s, namespaces::n},
FORALL_NS_SYMBOLS(SYMBOL_ENTRY)
#undef SYMBOL_ENTRY
};
} // namespace
InternedStrings::InternedStrings()
: sym_to_info_(static_cast<size_t>(_keys::num_symbols)) {
// Instead of a loop, this could be done by expanding the
// assignments directly into FORALL_NS_SYMBOLS, but it would create
// a huge function (thanks to all the std::string constructors and
// operator[]s) which would take several minutes to optimize. A
// static C array of constexpr-constructible structs takes instead
// no time to compile.
for (const auto& entry : entries) {
string_to_sym_[entry.qual_name] = entry.sym;
sym_to_info_[entry.sym] = {
entry.ns_sym, entry.qual_name, entry.unqual_name};
}
}
} // namespace c10