From f7bfe5e7efbb9d04b1297f404de6fd76a6a07f4c Mon Sep 17 00:00:00 2001
From: Bhavya Medishetty <86327865+bmedishe@users.noreply.github.com>
Date: Tue, 18 Apr 2023 13:37:25 -0500
Subject: [PATCH] [ROCm] temporary workaround till __double2half support
 enabled in HIP (#3236)

* temporary WAR workaround till __double2half support enabled in HIP

* workaround only for hipcc

---------

Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
---
 csrc/includes/conversion_utils.h | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/csrc/includes/conversion_utils.h b/csrc/includes/conversion_utils.h
index 27600b83..65fa6fd4 100644
--- a/csrc/includes/conversion_utils.h
+++ b/csrc/includes/conversion_utils.h
@@ -266,7 +266,12 @@ DS_D_INLINE float2 to(__nv_bfloat162 val)
 template <>
 DS_D_INLINE __half to(double val)
 {
+#ifdef __HIP_PLATFORM_HCC__
+    float val_f = __double2float_rn(val);
+    return __float2half(val_f);
+#else
     return __double2half(val);
+#endif
 }
 template <>
 DS_D_INLINE __half to(float val)
-- 
GitLab