include/hip/hip_fp16.h Source File

• include
• hip

hip_fp16.h

Go to the documentation of this file.

1/*

2 * Copyright (c) Advanced Micro Devices, Inc., or its affiliates.

3 *

4 * SPDX-License-Identifier: MIT

5 */

6

7#ifndef HIP_INCLUDE_HIP_HIP_FP16_H

8#define HIP_INCLUDE_HIP_HIP_FP16_H

9

10#include <hip/hip_common.h>

11

12#if defined(__HIP_PLATFORM_AMD__) && !defined(__HIP_PLATFORM_NVIDIA__)

13#include <hip/amd_detail/amd_hip_fp16.h>

14#elif !defined(__HIP_PLATFORM_AMD__) && defined(__HIP_PLATFORM_NVIDIA__)

15#define HIPRT_INF_FP16 CUDART_INF_FP16

16#define HIPRT_MAX_NORMAL_FP16 CUDART_MAX_NORMAL_FP16

17#define HIPRT_MIN_DENORM_FP16 CUDART_MIN_DENORM_FP16

18#define HIPRT_NAN_FP16 CUDART_NAN_FP16

19#define HIPRT_NEG_ZERO_FP16 CUDART_NEG_ZERO_FP16

20#define HIPRT_ONE_FP16 CUDART_ONE_FP16

21#define HIPRT_ZERO_FP16 CUDART_ZERO_FP16

22

23#include "cuda_fp16.h"

24#else

25#error ("Must define exactly one of __HIP_PLATFORM_AMD__ or __HIP_PLATFORM_NVIDIA__");

26#endif

27

28#endif

hip_common.h