Skip to content

Commit 7af61db

Browse files
committed
typo
1 parent 9fcc4d8 commit 7af61db

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

csrc/cuda/utils.cuh

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,13 +6,10 @@
66
AT_ASSERTM(x.device().is_cuda(), #x " must be CUDA tensor")
77
#define CHECK_INPUT(x) AT_ASSERTM(x, "Input mismatch")
88

9-
#ifndef _WIN32
10-
__device__ __inline__ at::Half __shfl_sync(const unsigned mask,
11-
const at::Half var,
12-
const unsigned int srcLane) {
9+
__device__ __inline__ at::Half
10+
__shfl_sync(const unsigned mask, const at::Half var, const int srcLane) {
1311
return __shfl_sync(mask, (__half)var, srcLane);
1412
}
15-
#endif
1613

1714
__device__ __inline__ at::Half __shfl_down_sync(const unsigned mask,
1815
const at::Half var,

0 commit comments

Comments
 (0)