Graphite
Loading...
Searching...
No Matches
types.hpp File Reference
#include <cuda_bf16.h>
#include <cuda_fp16.h>
Include dependency graph for types.hpp:
This graph shows which files directly or indirectly include this file:

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)