Update deform_conv_cuda.cu

fix issue 38: KaihuaTang/Scene-Graph-Benchmark.pytorch#38
moonshot-s · Jul 18, 2020 · dc5809d · dc5809d
1 parent 96eab8c
commit dc5809d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/maskrcnn_benchmark/csrc/cuda/deform_conv_cuda.cu b/maskrcnn_benchmark/csrc/cuda/deform_conv_cuda.cu
@@ -1,6 +1,10 @@
 // modify from
 // https://github.com/chengdazhi/Deformable-Convolution-V2-PyTorch/blob/mmdetection/mmdet/ops/dcn/src/deform_conv_cuda.c
 
+#ifndef AT_CHECK
+#define AT_CHECK TORCH_CHECK 
+#endif
+
 #include <ATen/ATen.h>
 #include <ATen/cuda/CUDAContext.h>