From cabb17afdfec90cdc9d17bd4a3f7ec41f688e011 Mon Sep 17 00:00:00 2001 From: Jatin Chaudhary Date: Mon, 24 Jun 2024 14:35:23 +0100 Subject: [PATCH] SWDEV-466747 - Add alias of HIPRT macro in CUDA Change-Id: I9bc6ba3b05c5e45d075446c53511e079f5d0fbcb [ROCm/hipother commit: 0ba07a685e26854dc173dbf2b5738a5f789c495d] --- .../hipnv/include/hip/nvidia_detail/nvidia_hip_bf16.h | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_bf16.h b/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_bf16.h index 118996af1d..163007facf 100644 --- a/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_bf16.h +++ b/projects/hipother/hipnv/include/hip/nvidia_detail/nvidia_hip_bf16.h @@ -20,13 +20,20 @@ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ - #ifndef HIP_INCLUDE_HIP_NVIDIA_DETAIL_HIP_FP16_H #define HIP_INCLUDE_HIP_NVIDIA_DETAIL_HIP_FP16_H +#define HIPRT_ONE_BF16 CUDART_ONE_BF16 +#define HIPRT_ZERO_BF16 CUDART_ZERO_BF16 +#define HIPRT_INF_BF16 CUDART_INF_BF16 +#define HIPRT_MAX_NORMAL_BF16 CUDART_MAX_NORMAL_BF16 +#define HIPRT_MIN_DENORM_BF16 CUDART_MIN_DENORM_BF16 +#define HIPRT_NAN_BF16 CUDART_NAN_BF16 +#define HIPRT_NEG_ZERO_BF16 CUDART_NEG_ZERO_BF16 + #include typedef struct __nv_bfloat16 __hip_bfloat16; typedef struct __nv_bfloat162 __hip_bfloat162; -#endif // HIP_INCLUDE_HIP_NVIDIA_DETAIL_HIP_FP16_H +#endif // HIP_INCLUDE_HIP_NVIDIA_DETAIL_HIP_FP16_H