include/ck_tile/ops/gemm/warp/warp_gemm_dispatcher.hpp File Reference

include/ck_tile/ops/gemm/warp/warp_gemm_dispatcher.hpp File Reference#

Composable Kernel: 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::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 8, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 8, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 16, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 32, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 16, 16, 32, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 4, 64, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 64, 4, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 8, false, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::half_t, ck_tile::half_t, float, 32, 32, 16, false, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 8, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 8, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 16, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 32, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 16, 16, 32, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 4, 64, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 64, 4, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 8, false, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf16_t, ck_tile::bf16_t, float, 32, 32, 16, false, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 32, 32, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::fp8_t, ck_tile::fp8_t, float, 32, 32, 16, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 32, 32, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::fp8_t, ck_tile::bf8_t, float, 32, 32, 16, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 32, 32, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf8_t, ck_tile::fp8_t, float, 32, 32, 16, true >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 32, 32, 16, false >
 
struct  ck_tile::impl::WarpGemmMfmaDispatcher< ck_tile::bf8_t, ck_tile::bf8_t, float, 32, 32, 16, true >
 

Namespaces

 ck_tile
 
 ck_tile::impl
 

Typedefs

template<typename AType , typename BType , typename CType , index_t MPerWave, index_t NPerWave, index_t KPerWave, bool TransposeC, bool SwizzleA = false>
using ck_tile::WarpGemmMfmaDispatcher = typename impl::WarpGemmMfmaDispatcher< AType, BType, CType, MPerWave, NPerWave, KPerWave, TransposeC, SwizzleA >::Type