Pikaliov/DCN_custom_op
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for Windows compilation

Browse Source
This commit is contained in:
Haoming Tang
2024-04-12 11:31:15 +08:00
committed by GitHub
parent 3199a6509f
commit e8351f4ff0
1 changed files with 5 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
DCNv4_op/src/cuda/common.h
View File

@@ -18,6 +18,10 @@
#include <cuda_fp16.h> #include <cuda_fp16.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#ifdef _WIN32
#define uint unsigned int
#endif
constexpr int kWarpSize = 32; constexpr int kWarpSize = 32;
#define opmath_t at::opmath_type<scalar_t> #define opmath_t at::opmath_type<scalar_t>