|
Graphite
|
#include <cuda_bf16.h>#include <cuda_fp16.h>

Go to the source code of this file.
Classes | |
| struct | graphite::is_half_or_bfloat16< T > |
| struct | graphite::is_half_or_bfloat16< __half > |
| struct | graphite::is_half_or_bfloat16< __nv_bfloat16 > |
| struct | graphite::vec2_type< float > |
| struct | graphite::vec2_type< double > |
| struct | graphite::vec2_type< __half > |
| struct | graphite::vec2_type< __nv_bfloat16 > |
| struct | graphite::Empty |
Typedefs | |
| template<typename T > | |
| using | graphite::is_low_precision = is_half_or_bfloat16< T > |
| template<typename T , typename S > | |
| using | graphite::InvP = std::conditional_t< is_low_precision< S >::value, T, S > |
Functions | |
| template<typename hp , typename lp > | |
| __device__ lp | graphite::convert_to_low_precision (const hp &value) |