/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda/
/usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda
NameSizeModeActions
comm.h15250644editdlrm
device_set.h1520644editdlrm
Event.h4870644editdlrm
Module.h5030644editdlrm
nccl.h50370644editdlrm
override_macros.h18870644editdlrm
python_comm.h1140644editdlrm
python_nccl.h6160644editdlrm
restore_macros.h8200644editdlrm
serialization.h5240644editdlrm
Storage.h25850644editdlrm
Stream.h5240644editdlrm
THCP.h4210644editdlrm
undef_macros.h9320644editdlrm
utils.h8710644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/torch/csrc/cuda/device_set.h (152B)
#pragma once #include namespace torch { static constexpr size_t MAX_CUDA_DEVICES = 64; using device_set = std::bitset; }