/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hip/checkouts/docs-5.5.1/include/hip/hip_bfloat16.h Source File

• include
• hip

hip_bfloat16.h

Go to the documentation of this file.

1

29#ifndef _HIP_BFLOAT16_H_

30#define _HIP_BFLOAT16_H_

31

32#if (defined(__HIP_PLATFORM_HCC__) || defined(__HIP_PLATFORM_AMD__)) && !(defined(__HIP_PLATFORM_NVCC__) || defined(__HIP_PLATFORM_NVIDIA__))

33#include <hip/amd_detail/amd_hip_bfloat16.h>

34#elif !(defined(__HIP_PLATFORM_HCC__) || defined(__HIP_PLATFORM_AMD__)) && (defined(__HIP_PLATFORM_NVCC__) || defined(__HIP_PLATFORM_NVIDIA__))

35#warning "hip_bfloat16.h is not supported on nvidia platform"

36#else

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

38#endif

39

40#endif // _HIP_BFLOAT16_H_