pk_fp4.hpp File Reference

pk_fp4.hpp File Reference#

Composable Kernel: 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::detail::fp16x2_repr
struct  ck_tile::detail::bf16x2_repr
struct  ck_tile::detail::fp32x2_repr
struct  ck_tile::pk_float4_e2m1_t
struct  ck_tile::numeric_traits< pk_fp4_t >
struct  ck_tile::numeric< pk_fp4_t >

Namespaces

namespace  ck_tile
namespace  ck_tile::detail

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::type

Functions

CK_TILE_HOST_DEVICE constexpr _Float16 ck_tile::detail::lane0 (const fp16x2_t &v)
CK_TILE_HOST_DEVICE constexpr _Float16 ck_tile::detail::lane1 (const fp16x2_t &v)
CK_TILE_HOST_DEVICE constexpr bfloat16_t ck_tile::detail::lane0 (const bf16x2_t &v)
CK_TILE_HOST_DEVICE constexpr bfloat16_t ck_tile::detail::lane1 (const bf16x2_t &v)
CK_TILE_HOST_DEVICE constexpr float ck_tile::detail::lane0 (const fp32x2_t &v)
CK_TILE_HOST_DEVICE constexpr float ck_tile::detail::lane1 (const fp32x2_t &v)
CK_TILE_HOST_DEVICE constexpr pk_float4_e2m1_t ck_tile::float_to_pk_fp4 (const float &x, float scale=1.f)
CK_TILE_HOST_DEVICE constexpr pk_fp4_raw_t ck_tile::float_to_mxfp4 (float x, float scale)
CK_TILE_HOST_DEVICE constexpr pk_fp4_t ck_tile::fp16_to_pk_fp4 (const fp16_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr pk_fp4_t ck_tile::bf16_to_pk_fp4 (const bf16_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr pk_fp4_t ck_tile::fp16x2_to_pk_fp4 (const fp16x2_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr pk_fp4_t ck_tile::bf16x2_to_pk_fp4 (const bf16x2_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr pk_fp4_t ck_tile::fp32x2_to_pk_fp4 (const fp32x2_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr fp32x2_t ck_tile::pk_fp4_to_fp32x2 (const pk_fp4_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr fp16x2_t ck_tile::pk_fp4_to_fp16x2 (const pk_fp4_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr bf16x2_t ck_tile::pk_fp4_to_bf16x2 (const pk_fp4_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr float ck_tile::pk_fp4_to_float (const pk_fp4_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr fp16_t ck_tile::pk_fp4_to_fp16 (const pk_fp4_t &x, float scale)
CK_TILE_HOST_DEVICE constexpr 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