Update deform_pool_cuda.cu

fix issue 38: KaihuaTang/Scene-Graph-Benchmark.pytorch#38
moonshot-s · Jul 18, 2020 · bbef4d9 · bbef4d9
1 parent dc5809d
commit bbef4d9
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/maskrcnn_benchmark/csrc/cuda/deform_pool_cuda.cu b/maskrcnn_benchmark/csrc/cuda/deform_pool_cuda.cu
@@ -5,6 +5,10 @@
 // author: Charles Shang
 // https://github.com/torch/cunn/blob/master/lib/THCUNN/generic/SpatialConvolutionMM.cu
 
+#ifndef AT_CHECK
+#define AT_CHECK TORCH_CHECK 
+#endif
+
 #include <ATen/ATen.h>
 #include <ATen/cuda/CUDAContext.h>