diff --git a/examples/common/dist_gemm_helpers.h b/examples/common/dist_gemm_helpers.h index ef258e6922..8062b4436e 100644 --- a/examples/common/dist_gemm_helpers.h +++ b/examples/common/dist_gemm_helpers.h @@ -60,7 +60,9 @@ using AtomicBoolean = cuda::atomic; __global__ void delay_kernel(const AtomicBoolean* atomic_flag_ptr) { while (not atomic_flag_ptr->load()) { +#if defined(__CUDA_ARCH__) && __CUDA_ARCH__ >= 700 __nanosleep(40); +#endif } }