Merge pull request #42 from tanghaom/main
Add support for Windows compilation
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user