/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/kernel/fmha_bwd_kernel.hpp File Reference

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/kernel/fmha_bwd_kernel.hpp File Reference#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fmha/kernel/fmha_bwd_kernel.hpp File Reference
fmha_bwd_kernel.hpp File Reference
#include "ck_tile/core.hpp"
#include "ck_tile/ops/common.hpp"
#include "ck_tile/ops/fmha/block/block_attention_bias_enum.hpp"
#include "ck_tile/ops/fmha/pipeline/block_fmha_bwd_dq_dk_dv_pipeline_selector.hpp"
#include <string>
#include <type_traits>
#include <utility>
#include <variant>

Go to the source code of this file.

Classes

struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::t2s< ck_tile::fp16_t >
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::t2s< ck_tile::bf16_t >
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdEmptyKargs< I >
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdCommonKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdCommonBiasKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdBatchModeBiasKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdAlibiKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdCommonBiasGradKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdBatchModeBiasGradKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdMaskKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdDropoutSeedOffset
 
union  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdDropoutSeedOffset::ValueOrPointer< T >
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdCommonDropoutKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdBatchModeDropoutKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdDeterministicKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdBatchModeKargs
 
struct  ck_tile::FmhaBwdDQDKDVKernel< FmhaPipeline_, KGradEpiloguePipeline_, VGradEpiloguePipeline_, QGradEpiloguePipeline_ >::FmhaBwdGroupModeKargs
 
struct  ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >
 
struct  ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::t2s< ck_tile::fp16_t >
 
struct  ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::t2s< ck_tile::bf16_t >
 
struct  ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOCommonKargs
 
struct  ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOBatchModeKargs
 
struct  ck_tile::FmhaBwdOGradDotOKernel< FmhaBwdOGradDotO_ >::FmhaBwdOGradDotOGroupModeKargs
 
struct  ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >
 
struct  ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::t2s< ck_tile::fp16_t >
 
struct  ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::t2s< ck_tile::bf16_t >
 
struct  ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradEmptyKargs< I >
 
struct  ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradCommonKargs
 
struct  ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradDeterministicKargs
 
struct  ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradBatchModeKargs
 
struct  ck_tile::FmhaBwdConvertQGradKernel< FmhaBwdConvertQGrad_ >::FmhaBwdConvertQGradGroupModeKargs
 

Namespaces

 ck_tile
 

Macros

#define _SS_   std::string
 
#define _TS_   std::to_string
 
#define _SS_   std::string
 
#define _TS_   std::to_string
 
#define _SS_   std::string
 
#define _TS_   std::to_string
 

Macro Definition Documentation

◆ _SS_ [1/3]

#define _SS_   std::string

◆ _SS_ [2/3]

#define _SS_   std::string

◆ _SS_ [3/3]

#define _SS_   std::string

◆ _TS_ [1/3]

#define _TS_   std::to_string

◆ _TS_ [2/3]

#define _TS_   std::to_string

◆ _TS_ [3/3]

#define _TS_   std::to_string