FmhaBwdOGradDotOGroupModeKargs Struct Reference#
ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOGroupModeKargs Struct Reference
#include <fmha_bwd_kernel.hpp>
Inheritance diagram for ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOGroupModeKargs:
Public Attributes | |
| const int32_t * | seqstart_q_ptr |
| const int32_t * | seqlen_q_ptr |
| const int32_t * | cu_seqlen_q_ptr |
| Public Attributes inherited from ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs | |
| 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
◆ cu_seqlen_q_ptr
template<typename FmhaBwdOGradDotO_>
| const int32_t* ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOGroupModeKargs::cu_seqlen_q_ptr |
◆ seqlen_q_ptr
template<typename FmhaBwdOGradDotO_>
| const int32_t* ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOGroupModeKargs::seqlen_q_ptr |
◆ seqstart_q_ptr
template<typename FmhaBwdOGradDotO_>
| const int32_t* ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOGroupModeKargs::seqstart_q_ptr |
The documentation for this struct was generated from the following file: