Skip to content

Commit 7a6c9ab

Browse files
alepack89Alessandro Pacchioni
andauthored
used explicit at::Half to __half conversion operator to fix ambiguous conversion error (rusty1s#220)
Co-authored-by: Alessandro Pacchioni <apacchioni@ambarella.com>
1 parent 122811a commit 7a6c9ab

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
@@ -8,11 +8,11 @@
88

99
__device__ __inline__ at::Half
1010
__shfl_sync(const unsigned mask, const at::Half var, const int srcLane) {
11-
return __shfl_sync(mask, (__half)var, srcLane);
11+
return __shfl_sync(mask, var.operator __half(), srcLane);
1212
}
1313

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

0 commit comments

Comments
 (0)