torch/lib/c10d/test/CUDATest.hpp

#pragma once

#include <c10/cuda/CUDAStream.h>

namespace c10d {
namespace test {

void cudaSleep(at::cuda::CUDAStream& stream, uint64_t clocks);

int cudaNumDevices();

} // namespace test
} // namespace c10d