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