/usr/local/lib64/python3.6/site-packages/torch/include/THC
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/THC/THCSleep.h (232B)
#ifndef THC_SPIN_INC
#define THC_SPIN_INC
#include
#include
// enqueues a kernel that spins for the specified number of cycles
TORCH_CUDA_CU_API void THC_sleep(THCState* state, int64_t cycles);
#endif