From 12d18d0fee5bad2eb0b92f176b868e23b2cd3fb3 Mon Sep 17 00:00:00 2001
From: Alessandro Pacchioni <apacchioni@ambarella.com>
Date: Fri, 15 Apr 2022 15:23:39 +0200
Subject: [PATCH] used explicit at::Half to __half conversion operator to fix
 ambiguous conversion error

---
 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 4b98452a..c4ef0530 100644
--- a/csrc/cuda/utils.cuh
+++ b/csrc/cuda/utils.cuh
@@ -8,11 +8,11 @@
 
 __device__ __inline__ at::Half
 __shfl_sync(const unsigned mask, const at::Half var, const int srcLane) {
-  return __shfl_sync(mask, (__half)var, srcLane);
+  return __shfl_sync(mask, var.operator __half(), srcLane);
 }
 
 __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);
 }