/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/core/numeric/pk_fp4.hpp File Reference

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/core/numeric/pk_fp4.hpp File Reference#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/core/numeric/pk_fp4.hpp File Reference
Classes | Namespaces | Macros | Typedefs | Functions
pk_fp4.hpp File Reference
#include <cmath>
#include "ck_tile/core/config.hpp"
#include "ck_tile/core/numeric/half.hpp"
#include "ck_tile/core/numeric/mxfp_convert.hpp"

Go to the source code of this file.

Classes

struct  ck_tile::pk_float4_e2m1_t
 
struct  ck_tile::numeric_traits< pk_fp4_t >
 
struct  ck_tile::numeric< pk_fp4_t >
 

Namespaces

 ck_tile
 

Macros

#define CK_TILE_FP4_CVT_DEVICE   0
 
#define TEST_convert_with_table   0
 

Typedefs

using ck_tile::fp32_t = float
 
using ck_tile::fp32x2_t = float
 
using ck_tile::bf16x2_t = bfloat16_t
 
using ck_tile::pk_fp4_t = pk_float4_e2m1_t
 
using ck_tile::pk_fp4_raw_t = typename pk_fp4_t::raw_type
 

Functions

constexpr CK_TILE_HOST_DEVICE uint8_t ck_tile::float_to_e2m1 (float x, float scale=1.f)
 
constexpr CK_TILE_HOST_DEVICE pk_fp4_t ck_tile::float_to_pk_fp4 (const float &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE pk_fp4_t ck_tile::fp16_to_pk_fp4 (const fp16_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE pk_fp4_t ck_tile::bf16_to_pk_fp4 (const bf16_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE pk_fp4_t ck_tile::fp16x2_to_pk_fp4 (const fp16x2_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE pk_fp4_t ck_tile::bf16x2_to_pk_fp4 (const bf16x2_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE pk_fp4_t ck_tile::fp32x2_to_pk_fp4 (const fp32x2_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE fp32x2_t ck_tile::pk_fp4_to_fp32x2 (const pk_fp4_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE fp16x2_t ck_tile::pk_fp4_to_fp16x2 (const pk_fp4_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE bf16x2_t ck_tile::pk_fp4_to_bf16x2 (const pk_fp4_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE float ck_tile::pk_fp4_to_float (const pk_fp4_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE fp16_t ck_tile::pk_fp4_to_fp16 (const pk_fp4_t &x, float scale)
 
constexpr CK_TILE_HOST_DEVICE bf16_t ck_tile::pk_fp4_to_bf16 (const pk_fp4_t &x, float scale)
 

Macro Definition Documentation

◆ CK_TILE_FP4_CVT_DEVICE

#define CK_TILE_FP4_CVT_DEVICE   0

◆ TEST_convert_with_table

#define TEST_convert_with_table   0