FmhaBwdConvertQGradGroupModeKargs Struct Reference

FmhaBwdConvertQGradGroupModeKargs Struct Reference#

Composable Kernel: ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs Struct Reference
Public Attributes | List of all members
ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs Struct Reference

#include <fmha_bwd_kernel.hpp>

Inheritance diagram for ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs:
ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradCommonKargs

Public Attributes

const int32_t * seqstart_q_ptr
const int32_t * seqstart_k_ptr
const int32_t * seqlen_q_ptr
const int32_t * seqlen_k_ptr
const int32_t * cu_seqlen_q_ptr
const int32_t * cu_seqlen_k_ptr
Public Attributes inherited from ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradCommonKargs
const void * dq_acc_ptr
void * dq_ptr
ck_tile::index_t seqlen_q
ck_tile::index_t seqlen_k
ck_tile::index_t hdim_q
ck_tile::index_t stride_dq
ck_tile::index_t stride_dq_acc
ck_tile::index_t nhead_stride_dq
ck_tile::index_t nhead_stride_dq_acc

Member Data Documentation

◆ cu_seqlen_k_ptr

template<typename FmhaBwdConvertQGrad_>
const int32_t* ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs::cu_seqlen_k_ptr

◆ cu_seqlen_q_ptr

template<typename FmhaBwdConvertQGrad_>
const int32_t* ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs::cu_seqlen_q_ptr

◆ seqlen_k_ptr

template<typename FmhaBwdConvertQGrad_>
const int32_t* ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs::seqlen_k_ptr

◆ seqlen_q_ptr

template<typename FmhaBwdConvertQGrad_>
const int32_t* ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs::seqlen_q_ptr

◆ seqstart_k_ptr

template<typename FmhaBwdConvertQGrad_>
const int32_t* ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs::seqstart_k_ptr

◆ seqstart_q_ptr

template<typename FmhaBwdConvertQGrad_>
const int32_t* ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs::seqstart_q_ptr
The documentation for this struct was generated from the following file: