fix half conversion (#280)

19130401 · Matthias Fey · GitHub · ae1fce51 · 19130401
Unverified Commit 19130401 authored Mar 24, 2022 by Matthias Fey Committed by GitHub Mar 24, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

csrc/cuda/utils.cuh csrc/cuda/utils.cuh +2 -2

No files found.
--- a/csrc/cuda/utils.cuh
+++ b/csrc/cuda/utils.cuh
@@ -9,11 +9,11 @@
 __device__ __inline__ at::Half __shfl_up_sync(const unsigned mask,
                                              const at::Half var,
                                              const unsigned int delta) {
-  return __shfl_up_sync(mask, (__half)var, delta);
+  return __shfl_up_sync(mask, var.operator __half(), delta);
 }

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