FmhaBwdOGradDotOCommonKargs Struct Reference

FmhaBwdOGradDotOCommonKargs Struct Reference#

Composable Kernel: ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs Struct Reference
Public Attributes | List of all members
ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs Struct Reference

#include <fmha_bwd_kernel.hpp>

Inheritance diagram for ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs:
ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOBatchModeKargs ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOGroupModeKargs

Public Attributes

const void * o_ptr
const void * do_ptr
void * d_ptr
float p_undrop
ck_tile::index_t seqlen_q
ck_tile::index_t hdim_v
ck_tile::index_t stride_do
ck_tile::index_t stride_o
ck_tile::index_t nhead_stride_do
ck_tile::index_t nhead_stride_o
ck_tile::index_t nhead_stride_d

Member Data Documentation

◆ d_ptr

template<typename FmhaBwdOGradDotO_>
void* ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::d_ptr

◆ do_ptr

template<typename FmhaBwdOGradDotO_>
const void* ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::do_ptr

◆ hdim_v

template<typename FmhaBwdOGradDotO_>
ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::hdim_v

◆ nhead_stride_d

template<typename FmhaBwdOGradDotO_>
ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::nhead_stride_d

◆ nhead_stride_do

template<typename FmhaBwdOGradDotO_>
ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::nhead_stride_do

◆ nhead_stride_o

template<typename FmhaBwdOGradDotO_>
ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::nhead_stride_o

◆ o_ptr

template<typename FmhaBwdOGradDotO_>
const void* ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::o_ptr

◆ p_undrop

template<typename FmhaBwdOGradDotO_>
float ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::p_undrop

◆ seqlen_q

template<typename FmhaBwdOGradDotO_>
ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::seqlen_q

◆ stride_do

template<typename FmhaBwdOGradDotO_>
ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::stride_do

◆ stride_o

template<typename FmhaBwdOGradDotO_>
ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs::stride_o
The documentation for this struct was generated from the following file: