bfloat16.hpp File Reference

bfloat16.hpp File Reference#

Composable Kernel: bfloat16.hpp File Reference
bfloat16.hpp File Reference

Go to the source code of this file.

Classes

struct  ck_tile::numeric< bfloat16_t >
struct  ck_tile::numeric_traits< bfloat16_t >

Namespaces

namespace  ck_tile

Typedefs

using ck_tile::bfloat16_t = ushort
using ck_tile::bf16_t = bfloat16_t
using ck_tile::bf16_raw_t = uint16_t

Enumerations

enum class  ck_tile::bf16_rounding_mode {
  ck_tile::standard = 0 ,
  ck_tile::truncate_with_nan ,
  ck_tile::truncate ,
  ck_tile::standard_asm ,
  ck_tile::rta_asm
}

Functions

template<bf16_rounding_mode rounding = static_cast<bf16_rounding_mode>(CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT)>
CK_TILE_HOST_DEVICE constexpr uint16_t ck_tile::float_to_bf16_raw (float f, constant< rounding >={})
template<bf16_rounding_mode rounding = static_cast<bf16_rounding_mode>(CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT)>
CK_TILE_HOST_DEVICE constexpr uint16_t ck_tile::double_to_bf16_raw (double f, constant< rounding >={})
CK_TILE_HOST_DEVICE constexpr float ck_tile::bf16_to_float_raw (uint16_t x)
CK_TILE_HOST_DEVICE constexpr double ck_tile::bf16_to_double_raw (uint16_t x)
CK_TILE_HOST_DEVICE constexpr uint16_t ck_tile::float_to_bf16_rtn_raw (float f)
CK_TILE_HOST constexpr uint16_t ck_tile::float_to_bf16_rtn_asm (float f)
CK_TILE_HOST uint16_t ck_tile::float_to_bf16_rta_asm (float f)
CK_TILE_HOST_DEVICE constexpr uint16_t ck_tile::float_to_bf16_truc_nan_raw (float f)
CK_TILE_HOST_DEVICE constexpr uint16_t ck_tile::float_to_bf16_truc_raw (float f)
template<bf16_rounding_mode rounding = static_cast<bf16_rounding_mode>(CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT)>
CK_TILE_HOST_DEVICE constexpr bfloat16_t ck_tile::float_to_bf16 (float f, constant< rounding >={})
template<bf16_rounding_mode rounding = static_cast<bf16_rounding_mode>(CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT)>
CK_TILE_HOST_DEVICE constexpr bfloat16_t ck_tile::double_to_bf16 (double f, constant< rounding >={})
CK_TILE_HOST_DEVICE constexpr float ck_tile::bf16_to_float (bfloat16_t x)
CK_TILE_HOST_DEVICE constexpr double ck_tile::bf16_to_double (bfloat16_t x)
template<bf16_rounding_mode rounding = static_cast<bf16_rounding_mode>(CK_TILE_FLOAT_TO_BFLOAT16_DEFAULT)>
CK_TILE_HOST_DEVICE bfloat16_t constexpr ck_tile::fp16_to_bf16 (half_t f, constant< rounding >={})
CK_TILE_HOST_DEVICE constexpr half_t ck_tile::bf16_to_fp16 (bfloat16_t x)
CK_TILE_HOST_DEVICE bfloat16_t ck_tile::abs (const bfloat16_t &x)
CK_TILE_HOST_DEVICE bool ck_tile::isnan (const bfloat16_t &x)
CK_TILE_DEVICE bfloat16_t ck_tile::sqrt (bfloat16_t x)
CK_TILE_DEVICE bfloat16_t ck_tile::exp (bfloat16_t x)
CK_TILE_DEVICE bfloat16_t ck_tile::exp2 (bfloat16_t x)
CK_TILE_DEVICE bfloat16_t ck_tile::log (bfloat16_t x)