From 3c11b7734190c030b1b5961e4322459edf774966 Mon Sep 17 00:00:00 2001 From: rusty1s Date: Thu, 24 Mar 2022 12:44:19 +0000 Subject: [PATCH] fix half conversion --- csrc/cuda/utils.cuh | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/csrc/cuda/utils.cuh b/csrc/cuda/utils.cuh index d08b0536..ee2e9108 100644 --- 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); }