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

• develop
• include
• ck_tile
• ops
• gemm
• warp

Classes | Namespaces | Typedefs

warp_gemm_dispatcher.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm.hpp"
#include "ck_tile/ops/gemm/warp/warp_wmma_gemm.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< float, float, float, 16, 16, 4, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< float, float, float, 16, 16, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< float, float, float, 16, 16, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 8, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 8, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 16, false, false, false, EDouble >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 16, true, false, false, EDouble >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 16, 16, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 16, 16, 32, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 16, 16, 32, false, false, false, EDouble >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 16, 16, 32, true, false, false, EDouble >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 4, 64, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 64, 4, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 16, 16, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 16, 16, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 8, false, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 16, false, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 8, true, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 16, true, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 32, 32, 16, false, false, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< half_t, half_t, float, 16, 16, 32, false, false, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 8, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 8, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 16, false, false, false, EDouble >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 16, true, false, false, EDouble >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 16, 16, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 16, 16, 32, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 16, 16, 32, false, false, false, EDouble >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 16, 16, 32, true, false, false, EDouble >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 4, 64, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 64, 4, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 16, 16, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 16, 16, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 8, false, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 16, false, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 8, true, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf16_t, bf16_t, float, 32, 32, 16, true, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 32, 32, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 32, 32, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 16, 16, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 16, 16, 64, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 32, 32, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 16, 16, 32, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 32, 32, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 32, 32, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 16, 16, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 32, 32, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, fp8_t, float, 32, 32, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, fp8_t, float, 32, 32, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 32, 32, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 32, 32, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 16, 16, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 16, 16, 32, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 16, 16, 64, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 32, 32, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 16, 16, 128, false, false, false, I >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 16, 16, 128, false, false, false, I >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, fp8_t, float, 16, 16, 128, false, false, false, I >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 16, 16, 128, false, false, false, I >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 16, 16, 128, true, false, false, I >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 16, 16, 128, true, false, false, I >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, fp8_t, float, 16, 16, 128, true, false, false, I >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 16, 16, 128, true, false, false, I >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< pk_fp4_t, pk_fp4_t, float, 16, 16, 128, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 32, 32, 64, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 32, 32, 64, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, fp8_t, float, 32, 32, 64, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 32, 32, 64, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 32, 32, 64, false, false, false, EQuad >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 32, 32, 64, false, false, false, EQuad >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, fp8_t, float, 32, 32, 64, false, false, false, EQuad >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 32, 32, 64, false, false, false, EQuad >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, fp8_t, float, 16, 16, 16, TransposeC, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, bf8_t, float, 16, 16, 16, TransposeC, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< fp8_t, bf8_t, float, 16, 16, 16, TransposeC, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< bf8_t, fp8_t, float, 16, 16, 16, TransposeC, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< int8_t, int8_t, int32_t, 32, 32, 16, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< int8_t, int8_t, int32_t, 32, 32, 16, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< int8_t, int8_t, int32_t, 16, 16, 32, false >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< int8_t, int8_t, int32_t, 16, 16, 32, true >
struct	ck_tile::impl::warp_gemm_dispatcher::Dispatcher< int8_t, int8_t, int32_t, 16, 16, 16, TransposeC, false >

Namespaces
	ck_tile
	ck_tile::impl
	ck_tile::impl::warp_gemm_dispatcher

Typedefs
template<typename AType , typename BType , typename AccType , index_t MPerWave, index_t NPerWave, index_t KPerWave, bool TransposeC, bool SwizzleA = false, bool UseStructuredSparsity = false, WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmDispatcher = typename impl::warp_gemm_dispatcher::Dispatcher< AType, BType, AccType, MPerWave, NPerWave, KPerWave, TransposeC, SwizzleA, UseStructuredSparsity, AttrNumAccess >::Type