Repository URL to install this package:
#pragma once #include <c10/cuda/CUDAMacros.h> #include <bitset> #include <cstddef> namespace torch { using device_set = std::bitset<C10_COMPILE_TIME_MAX_GPUS>; } // namespace torch