"research/object_detection/g3doc/running_pets.md" did not exist on "1e093b26a5c5200856f90ad8ad9c794f2dd721fc"
cuda_helpers.h 475 Bytes
Newer Older
1
2
#pragma once

3
4
5
namespace vision {
namespace ops {

limm's avatar
limm committed
6
7
#define CUDA_1D_KERNEL_LOOP_T(i, n, index_t)                         \
  for (index_t i = (blockIdx.x * blockDim.x) + threadIdx.x; i < (n); \
8
       i += (blockDim.x * gridDim.x))
9

limm's avatar
limm committed
10
11
#define CUDA_1D_KERNEL_LOOP(i, n) CUDA_1D_KERNEL_LOOP_T(i, n, int)

12
template <typename integer>
13
constexpr __host__ __device__ inline integer ceil_div(integer n, integer m) {
14
15
  return (n + m - 1) / m;
}
16
17
18

} // namespace ops
} // namespace vision