/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

/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#

Composable Kernel: /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
warp_gemm_dispatcher.hpp File Reference

Go to the source code of this file.

Classes

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::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::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::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