We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent ae1fce5 commit 1913040Copy full SHA for 1913040
csrc/cuda/utils.cuh
@@ -9,11 +9,11 @@
9
__device__ __inline__ at::Half __shfl_up_sync(const unsigned mask,
10
const at::Half var,
11
const unsigned int delta) {
12
- return __shfl_up_sync(mask, (__half)var, delta);
+ return __shfl_up_sync(mask, var.operator __half(), delta);
13
}
14
15
__device__ __inline__ at::Half __shfl_down_sync(const unsigned mask,
16
17
18
- return __shfl_down_sync(mask, (__half)var, delta);
+ return __shfl_down_sync(mask, var.operator __half(), delta);
19
0 commit comments