Add support for Windows compilation

This commit is contained in:
Haoming Tang
2024-04-12 11:31:15 +08:00
committed by GitHub
parent 3199a6509f
commit e8351f4ff0

View File

@@ -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
#endif