1 | #include <parallel_type_bitmap.h> |
---|---|
2 | |
3 | namespace torch { |
4 | namespace jit { |
5 | namespace fuser { |
6 | namespace cuda { |
7 | |
8 | constexpr std::bitset<ParallelTypeBitmap::kNumParallelTypes> |
9 | ParallelTypeBitmap::kTIDBits; |
10 | constexpr std::bitset<ParallelTypeBitmap::kNumParallelTypes> |
11 | ParallelTypeBitmap::kBIDBits; |
12 | |
13 | std::string ParallelTypeBitmap::toString() const { |
14 | std::stringstream ss; |
15 | ss << "("; |
16 | bool is_first = true; |
17 | for (ParallelType pt : *this) { |
18 | if (!is_first) { |
19 | ss << " "; |
20 | } |
21 | ss << pt; |
22 | is_first = false; |
23 | } |
24 | ss << ")"; |
25 | return ss.str(); |
26 | } |
27 | |
28 | } // namespace cuda |
29 | } // namespace fuser |
30 | } // namespace jit |
31 | } // namespace torch |
32 |