There was an error while loading. Please reload this page.
1 parent 122811a commit 7a6c9abCopy full SHA for 7a6c9ab
csrc/cuda/utils.cuh
@@ -8,11 +8,11 @@
8
9
__device__ __inline__ at::Half
10
__shfl_sync(const unsigned mask, const at::Half var, const int srcLane) {
11
- return __shfl_sync(mask, (__half)var, srcLane);
+ return __shfl_sync(mask, var.operator __half(), srcLane);
12
}
13
14
__device__ __inline__ at::Half __shfl_down_sync(const unsigned mask,
15
const at::Half var,
16
const unsigned int delta) {
17
- return __shfl_down_sync(mask, (__half)var, delta);
+ return __shfl_down_sync(mask, var.operator __half(), delta);
18
0 commit comments