AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-501779 - add macro alias for hip for fp16

Browse Source 
Change-Id: I3854c7e1b7728eb959b25224db702ea5e4425c2e
This commit is contained in:
Jatin Chaudhary
2024-12-04 09:25:43 +00:00
committed by Jatin Jaikishan Chaudhary
parent 808274984d
commit c1f7109cdd
1 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/hip/hip_fp16.h
+8
View File
@@ -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__");