Rmsnorm2dFwdHostArgs Struct Reference

Rmsnorm2dFwdHostArgs Struct Reference#

Composable Kernel: ck_tile::Rmsnorm2dFwdHostArgs Struct Reference
Public Attributes | List of all members
ck_tile::Rmsnorm2dFwdHostArgs Struct Reference

#include <rmsnorm2d_fwd_kernel.hpp>

Public Attributes

const void * p_x
const void * p_x_residual
const void * p_sm_scale
const void * p_gamma
void * p_y
void * p_y_residual
void * p_y_scale
void * p_invRms
void * p_y_unquant
float epsilon
index_t m
index_t n
index_t x_stride
index_t xr_stride
index_t y_stride
index_t yr_stride

Member Data Documentation

◆ epsilon

float ck_tile::Rmsnorm2dFwdHostArgs::epsilon

◆ m

index_t ck_tile::Rmsnorm2dFwdHostArgs::m

◆ n

index_t ck_tile::Rmsnorm2dFwdHostArgs::n

◆ p_gamma

const void* ck_tile::Rmsnorm2dFwdHostArgs::p_gamma

◆ p_invRms

void* ck_tile::Rmsnorm2dFwdHostArgs::p_invRms

◆ p_sm_scale

const void* ck_tile::Rmsnorm2dFwdHostArgs::p_sm_scale

◆ p_x

const void* ck_tile::Rmsnorm2dFwdHostArgs::p_x

◆ p_x_residual

const void* ck_tile::Rmsnorm2dFwdHostArgs::p_x_residual

◆ p_y

void* ck_tile::Rmsnorm2dFwdHostArgs::p_y

◆ p_y_residual

void* ck_tile::Rmsnorm2dFwdHostArgs::p_y_residual

◆ p_y_scale

void* ck_tile::Rmsnorm2dFwdHostArgs::p_y_scale

◆ p_y_unquant

void* ck_tile::Rmsnorm2dFwdHostArgs::p_y_unquant

◆ x_stride

index_t ck_tile::Rmsnorm2dFwdHostArgs::x_stride

◆ xr_stride

index_t ck_tile::Rmsnorm2dFwdHostArgs::xr_stride

◆ y_stride

index_t ck_tile::Rmsnorm2dFwdHostArgs::y_stride

◆ yr_stride

index_t ck_tile::Rmsnorm2dFwdHostArgs::yr_stride
The documentation for this struct was generated from the following file: