Skip to content

Commit 3a161f2

Browse files
mxz297facebook-github-bot
authored andcommitted
hipify detectron2
Summary: As title Differential Revision: D76601224 fbshipit-source-id: b677952b563d1f404855bd6a160568592994d1aa
1 parent 65184fc commit 3a161f2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

detectron2/layers/csrc/deformable/deform_conv_cuda_kernel.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,12 +73,12 @@
7373
#include <float.h>
7474
#include <math.h>
7575
#include <stdio.h>
76-
#include <THC/THCAtomics.cuh>
76+
#include <ATen/cuda/Atomic.cuh>
7777

7878
using namespace at;
7979

8080
#define CUDA_KERNEL_LOOP(i, n) \
81-
for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < (n); \
81+
for (auto i = blockIdx.x * blockDim.x + threadIdx.x; i < (n); \
8282
i += blockDim.x * gridDim.x)
8383

8484

0 commit comments

Comments
 (0)