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(__HIPCC_RTC__)

33#include <hip/hip_common.h>

34#endif

35

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

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

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

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

40#else

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

42#endif

43

44#endif // _HIP_BFLOAT16_H_

hip_common.h