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