From c1f7109cdd0e7921403cea649baf24a3c38cdd20 Mon Sep 17 00:00:00 2001
From: Jatin Chaudhary <JatinJaikishan.Chaudhary@amd.com>
Date: Wed, 4 Dec 2024 09:25:43 +0000
Subject: [PATCH] SWDEV-501779 - add macro alias for hip for fp16

Change-Id: I3854c7e1b7728eb959b25224db702ea5e4425c2e
---
 include/hip/hip_fp16.h | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/include/hip/hip_fp16.h b/include/hip/hip_fp16.h
index 67def4ee1a..088225e6e9 100644
--- a/include/hip/hip_fp16.h
+++ b/include/hip/hip_fp16.h
@@ -28,6 +28,14 @@ THE SOFTWARE.
 #if defined(__HIP_PLATFORM_AMD__) && !defined(__HIP_PLATFORM_NVIDIA__)
 #include <hip/amd_detail/amd_hip_fp16.h>
 #elif !defined(__HIP_PLATFORM_AMD__) && defined(__HIP_PLATFORM_NVIDIA__)
+#define HIPRT_INF_FP16 CUDART_INF_FP16
+#define HIPRT_MAX_NORMAL_FP16 CUDART_MAX_NORMAL_FP16
+#define HIPRT_MIN_DENORM_FP16 CUDART_MIN_DENORM_FP16
+#define HIPRT_NAN_FP16 CUDART_NAN_FP16
+#define HIPRT_NEG_ZERO_FP16 CUDART_NEG_ZERO_FP16
+#define HIPRT_ONE_FP16 CUDART_ONE_FP16
+#define HIPRT_ZERO_FP16 CUDART_ZERO_FP16
+
 #include "cuda_fp16.h"
 #else
 #error("Must define exactly one of __HIP_PLATFORM_AMD__ or __HIP_PLATFORM_NVIDIA__");