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 #include +#ifdef _WIN32 +#define uint unsigned int +#endif + constexpr int kWarpSize = 32; #define opmath_t at::opmath_type @@ -209,4 +213,4 @@ __device__ void ms_deform_attn_col2im_bilinear( -#endif \ No newline at end of file +#endif