Kargs Struct Reference

• ck_tile
• Rmsnorm2dFwd
• Kargs

Public Attributes | List of all members

ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs 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

template<typename Pipeline_, typename Epilogue_>

float ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::epsilon

◆ m

template<typename Pipeline_, typename Epilogue_>

index_t ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::m

◆ n

template<typename Pipeline_, typename Epilogue_>

index_t ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::n

◆ p_gamma

template<typename Pipeline_, typename Epilogue_>

const void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_gamma

◆ p_invRms

template<typename Pipeline_, typename Epilogue_>

void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_invRms

◆ p_sm_scale

template<typename Pipeline_, typename Epilogue_>

const void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_sm_scale

◆ p_x

template<typename Pipeline_, typename Epilogue_>

const void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_x

◆ p_x_residual

template<typename Pipeline_, typename Epilogue_>

const void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_x_residual

◆ p_y

template<typename Pipeline_, typename Epilogue_>

void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_y

◆ p_y_residual

template<typename Pipeline_, typename Epilogue_>

void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_y_residual

◆ p_y_scale

template<typename Pipeline_, typename Epilogue_>

void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_y_scale

◆ p_y_unquant

template<typename Pipeline_, typename Epilogue_>

void* ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_y_unquant

◆ x_stride

template<typename Pipeline_, typename Epilogue_>

index_t ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::x_stride

◆ xr_stride

template<typename Pipeline_, typename Epilogue_>

index_t ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::xr_stride

◆ y_stride

template<typename Pipeline_, typename Epilogue_>

index_t ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::y_stride

◆ yr_stride

template<typename Pipeline_, typename Epilogue_>

index_t ck_tile::Rmsnorm2dFwd< Pipeline_, Epilogue_ >::Kargs::yr_stride

---

The documentation for this struct was generated from the following file:

• rmsnorm2d_fwd_kernel.hpp