1 #pragma once 2 3 #include <c10/cuda/CUDAMacros.h> 4 #include <bitset> 5 #include <cstddef> 6 7 namespace torch { 8 9 using device_set = std::bitset<C10_COMPILE_TIME_MAX_GPUS>; 10 11 } // namespace torch 12