Sleep.h 237 B

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