Kargs Struct Reference

Kargs Struct Reference#

Composable Kernel: ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs Struct Reference
Public Attributes | List of all members
ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs Struct Reference

#include <layernorm2d_fwd_kernel.hpp>

Public Attributes

const void * p_x
const void * p_x_residual
const void * p_sm_scale
const void * p_x_bias
const void * p_gamma
const void * p_beta
void * p_y
void * p_y_residual
void * p_y_scale
void * p_mean
void * p_invStd
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::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::epsilon

◆ m

template<typename Pipeline_, typename Epilogue_>
index_t ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::m

◆ n

template<typename Pipeline_, typename Epilogue_>
index_t ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::n

◆ p_beta

template<typename Pipeline_, typename Epilogue_>
const void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_beta

◆ p_gamma

template<typename Pipeline_, typename Epilogue_>
const void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_gamma

◆ p_invStd

template<typename Pipeline_, typename Epilogue_>
void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_invStd

◆ p_mean

template<typename Pipeline_, typename Epilogue_>
void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_mean

◆ p_sm_scale

template<typename Pipeline_, typename Epilogue_>
const void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_sm_scale

◆ p_x

template<typename Pipeline_, typename Epilogue_>
const void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_x

◆ p_x_bias

template<typename Pipeline_, typename Epilogue_>
const void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_x_bias

◆ p_x_residual

template<typename Pipeline_, typename Epilogue_>
const void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_x_residual

◆ p_y

template<typename Pipeline_, typename Epilogue_>
void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_y

◆ p_y_residual

template<typename Pipeline_, typename Epilogue_>
void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_y_residual

◆ p_y_scale

template<typename Pipeline_, typename Epilogue_>
void* ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::p_y_scale

◆ x_stride

template<typename Pipeline_, typename Epilogue_>
index_t ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::x_stride

◆ xr_stride

template<typename Pipeline_, typename Epilogue_>
index_t ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::xr_stride

◆ y_stride

template<typename Pipeline_, typename Epilogue_>
index_t ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::y_stride

◆ yr_stride

template<typename Pipeline_, typename Epilogue_>
index_t ck_tile::Layernorm2dFwd< Pipeline_, Epilogue_ >::Kargs::yr_stride
The documentation for this struct was generated from the following file: