Skip to content

Commit 1913040

Browse files
authored
fix half conversion (#280)
1 parent ae1fce5 commit 1913040

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

csrc/cuda/utils.cuh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,11 +9,11 @@
99
__device__ __inline__ at::Half __shfl_up_sync(const unsigned mask,
1010
const at::Half var,
1111
const unsigned int delta) {
12-
return __shfl_up_sync(mask, (__half)var, delta);
12+
return __shfl_up_sync(mask, var.operator __half(), delta);
1313
}
1414

1515
__device__ __inline__ at::Half __shfl_down_sync(const unsigned mask,
1616
const at::Half var,
1717
const unsigned int delta) {
18-
return __shfl_down_sync(mask, (__half)var, delta);
18+
return __shfl_down_sync(mask, var.operator __half(), delta);
1919
}

0 commit comments

Comments
 (0)