From dc5809d56e479e1e6c4b01e69ba0ce3bc9f79fc1 Mon Sep 17 00:00:00 2001
From: Karim-53 <mohamedkarim.belaid@supcom.tn>
Date: Sat, 18 Jul 2020 20:27:52 +0200
Subject: [PATCH] Update deform_conv_cuda.cu

fix issue 38: https://github.com/KaihuaTang/Scene-Graph-Benchmark.pytorch/issues/38
---
 maskrcnn_benchmark/csrc/cuda/deform_conv_cuda.cu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/maskrcnn_benchmark/csrc/cuda/deform_conv_cuda.cu b/maskrcnn_benchmark/csrc/cuda/deform_conv_cuda.cu
index 74f7d33..1c07c82 100644
--- 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>