1 #include <c10/macros/Macros.h> 2 #include <cstdint> 3 4 namespace at::cuda { 5 namespace detail { 6 void init_p2p_access_cache(int64_t num_devices); 7 } 8 9 TORCH_CUDA_CPP_API bool get_p2p_access(int source_dev, int dest_dev); 10 11 } // namespace at::cuda 12