include/hip/hip_bf16.h Source File

• include
• hip

hip_bf16.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_BF16_H

8#define HIP_INCLUDE_HIP_HIP_BF16_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_bf16.h>

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

15#include <hip/nvidia_detail/nvidia_hip_bf16.h>

16#else

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

18#endif

19

20#endif // HIP_INCLUDE_HIP_HIP_BF16_H

hip_common.h