[bb7f56]: / custom_extensions / nms / src / cuda_helpers.h

Download this file

6 lines (4 with data), 186 Bytes

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))