FmhaBwdOGradDotOBatchModeKargs Struct Reference#
ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOBatchModeKargs Struct Reference
#include <fmha_bwd_kernel.hpp>
Inheritance diagram for ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOBatchModeKargs:
Public Attributes | |
| ck_tile::index_t | batch_stride_do |
| ck_tile::index_t | batch_stride_o |
| ck_tile::index_t | batch_stride_d |
| 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
◆ batch_stride_d
template<typename FmhaBwdOGradDotO_>
| ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOBatchModeKargs::batch_stride_d |
◆ batch_stride_do
template<typename FmhaBwdOGradDotO_>
| ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOBatchModeKargs::batch_stride_do |
◆ batch_stride_o
template<typename FmhaBwdOGradDotO_>
| ck_tile::index_t ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOBatchModeKargs::batch_stride_o |
The documentation for this struct was generated from the following file: