1 | #pragma once |
---|---|
2 | |
3 | #include <bitset> |
4 | |
5 | namespace torch { |
6 | |
7 | static constexpr size_t MAX_CUDA_DEVICES = 64; |
8 | using device_set = std::bitset<MAX_CUDA_DEVICES>; |
9 | |
10 | } // namespace torch |
11 |
1 | #pragma once |
---|---|
2 | |
3 | #include <bitset> |
4 | |
5 | namespace torch { |
6 | |
7 | static constexpr size_t MAX_CUDA_DEVICES = 64; |
8 | using device_set = std::bitset<MAX_CUDA_DEVICES>; |
9 | |
10 | } // namespace torch |
11 |