FmhaBwdBatchModeBiasGradKargs Struct Reference

FmhaBwdBatchModeBiasGradKargs Struct Reference#

Composable Kernel: ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_ >::FmhaBwdBatchModeBiasGradKargs Struct Reference
ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_ >::FmhaBwdBatchModeBiasGradKargs Struct Reference

#include <fmha_bwd_kernel.hpp>

Inheritance diagram for ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_ >::FmhaBwdBatchModeBiasGradKargs:
ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_ >::FmhaBwdCommonBiasGradKargs

Public Attributes

ck_tile::index_t batch_stride_dbias = 0
 
- Public Attributes inherited from ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_ >::FmhaBwdCommonBiasGradKargs
void * dbias_ptr = nullptr
 
ck_tile::index_t stride_dbias = 0
 
ck_tile::index_t nhead_stride_dbias = 0
 

Member Data Documentation

◆ batch_stride_dbias

template<typename FmhaPipeline_ , typename KGradEpiloguePipeline_ , typename VGradEpiloguePipeline_ >
ck_tile::index_t ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_ >::FmhaBwdBatchModeBiasGradKargs::batch_stride_dbias = 0

The documentation for this struct was generated from the following file:
  • /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-7.0.1/include/ck_tile/ops/fmha/kernel/fmha_bwd_kernel.hpp