From e8351f4ff0fbc8ecdfbb9aafd03b0717e6024833 Mon Sep 17 00:00:00 2001
From: Haoming Tang <48461500+tanghaom@users.noreply.github.com>
Date: Fri, 12 Apr 2024 11:31:15 +0800
Subject: [PATCH] Add support for Windows compilation

---
 DCNv4_op/src/cuda/common.h | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/DCNv4_op/src/cuda/common.h b/DCNv4_op/src/cuda/common.h
index bf8f803..2134a40 100644
--- a/DCNv4_op/src/cuda/common.h
+++ b/DCNv4_op/src/cuda/common.h
@@ -18,6 +18,10 @@
 #include <cuda_fp16.h>
 #include <cuda_runtime.h>
 
+#ifdef _WIN32
+#define uint unsigned int
+#endif
+
 constexpr int kWarpSize = 32;
 #define opmath_t at::opmath_type<scalar_t>
 
@@ -209,4 +213,4 @@ __device__ void ms_deform_attn_col2im_bilinear(
 
 
 
-#endif
\ No newline at end of file
+#endif