HIP: Heterogenous-computing Interface for Portability
Loading...
Searching...
No Matches
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_