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:
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
Public Attributes inherited from