1 | #pragma once |
---|---|
2 | #include <c10/macros/Export.h> |
3 | #include <cstdint> |
4 | |
5 | namespace at { |
6 | namespace cuda { |
7 | |
8 | // enqueues a kernel that spins for the specified number of cycles |
9 | TORCH_CUDA_CU_API void sleep(int64_t cycles); |
10 | |
11 | }} // namespace at::cuda |
12 |