Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "dc9c9784dd7aa2917e6d9ed10a226b041defa256"
Switch branch/tag
Find file Normal viewHistoryPermalink
cuda_helpers.h 186 Bytes
Newer Older
Francisco Massa's avatar
Add C++ ops to torchvision (#826)  
Francisco Massa committed
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))