Merge pull request #42 from tanghaom/main

Add support for Windows compilation
This commit is contained in:
Yuwen Xiong
2024-04-13 00:23:09 +08:00
committed by GitHub

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