cuda_helpers.h 321 Bytes
Newer Older
1
2
3
4
5
#pragma once

#define CUDA_1D_KERNEL_LOOP(i, n)                                \
  for (int i = (blockIdx.x * blockDim.x) + threadIdx.x; i < (n); \
       i += (blockDim.x * gridDim.x))
6
7

template <typename integer>
8
constexpr __host__ __device__ inline integer ceil_div(integer n, integer m) {
9
10
  return (n + m - 1) / m;
}