/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::WarpGemmDispatcher< float, float, float, 16, 16, 4, false >
struct	ck_tile::impl::WarpGemmDispatcher< float, float, float, 16, 16, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< float, float, float, 16, 16, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 8, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 8, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, false, false, false, WGAttrNumAccessEnum::Double >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, true, false, false, WGAttrNumAccessEnum::Double >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 32, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 32, false, false, false, WGAttrNumAccessEnum::Double >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 32, true, false, false, WGAttrNumAccessEnum::Double >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 4, 64, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 64, 4, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 8, false, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, false, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 8, true, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, true, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, false, false, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 32, false, false, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 8, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 8, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, false, false, false, WGAttrNumAccessEnum::Double >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, true, false, false, WGAttrNumAccessEnum::Double >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 32, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 32, false, false, false, WGAttrNumAccessEnum::Double >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 32, true, false, false, WGAttrNumAccessEnum::Double >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 4, 64, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 64, 4, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 8, false, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, false, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 8, true, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, true, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 32, 32, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 32, 32, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 16, 16, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 16, 16, 64, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 32, 32, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 16, 16, 32, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 32, 32, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 32, 32, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 16, 16, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 32, 32, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 32, 32, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 32, 32, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 32, 32, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 32, 32, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 16, 16, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 16, 16, 32, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 16, 16, 64, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 32, 32, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 16, 16, 128, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 16, 16, 128, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 16, 16, 128, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 16, 16, 128, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 16, 16, 128, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 16, 16, 128, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 16, 16, 128, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 16, 16, 128, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::pk_fp4_t, ck_tile::pk_fp4_t, float, 16, 16, 128, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 32, 32, 64, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 32, 32, 64, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 32, 32, 64, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 32, 32, 64, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 32, 32, 64, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 32, 32, 64, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 32, 32, 64, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 32, 32, 64, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 16, 16, 128, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 16, 16, 128, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 16, 16, 128, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 16, 16, 128, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::pk_fp4_t, ck_tile::pk_fp4_t, float, 16, 16, 128, false, false, false, WGAttrNumAccessEnum::Quad >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 16, 16, 16, TransposeC, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 16, 16, 16, TransposeC, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 16, 16, 16, TransposeC, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 16, 16, 16, TransposeC, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::int8_t, ck_tile::int8_t, ck_tile::int32_t, 32, 32, 16, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::int8_t, ck_tile::int8_t, ck_tile::int32_t, 32, 32, 16, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::int8_t, ck_tile::int8_t, ck_tile::int32_t, 16, 16, 32, false >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::int8_t, ck_tile::int8_t, ck_tile::int32_t, 16, 16, 32, true >
struct	ck_tile::impl::WarpGemmDispatcher< ck_tile::int8_t, ck_tile::int8_t, int32_t, 16, 16, 16, TransposeC, false >

Namespaces
	ck_tile
	ck_tile::impl

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::WarpGemmDispatcher< AType, BType, AccType, MPerWave, NPerWave, KPerWave, TransposeC, SwizzleA, UseStructuredSparsity, AttrNumAccess >::Type