[skip ci]

Author: rusty1s

.github/workflows/building.yml

@@ -10,7 +10,8 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        os: [ubuntu-16.04, macos-10.15, windows-latest]
+        # os: [ubuntu-16.04, macos-10.15, windows-latest]
+        os: [windows-latest]
         python-version: [3.6, 3.7, 3.8, 3.9]
         torch-version: [1.8.0, 1.9.0]
         cuda-version: ['cpu', 'cu101', 'cu102', 'cu111']

csrc/cuda/utils.cuh

@@ -6,11 +6,13 @@
   AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
 #define CHECK_INPUT(x) AT_ASSERTM(x, "Input mismatch")
 
+#ifndef _WIN32
 __device__ __inline__ at::Half __shfl_sync(const unsigned mask,
                                            const at::Half var,
                                            const unsigned int srcLane) {
   return __shfl_sync(mask, (__half)var, srcLane);
 }
+#endif
 
 __device__ __inline__ at::Half __shfl_down_sync(const unsigned mask,
                                                 const at::Half var,