/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hip/checkouts/docs-5.5.0/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_