warp_gemm.hpp File Reference

• include
• ck_tile
• ops
• gemm
• warp

Namespaces | Typedefs

warp_gemm.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_impl.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_attribute_mfma.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_smfmac_impl.hpp"
#include "ck_tile/ops/gemm/warp/warp_gemm_attribute_smfmac.hpp"

Go to the source code of this file.

Namespaces
namespace	ck_tile

Typedefs
using	ck_tile::WarpGemmMfmaF32F32F32M16N16K4
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaF32F32F32M16N16K16
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaF32F32F32M16N16K16TransposedCDistribution
using	ck_tile::WarpGemmMfmaF16F16F32M32N32K8
using	ck_tile::WarpGemmMfmaF16F16F32M16N16K16
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaF16F16F32M32N32K16
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaF16F16F32M16N16K32
using	ck_tile::WarpGemmMfmaF16F16F32M32N32K8SwizzleA
using	ck_tile::WarpGemmMfmaF16F16F32M32N32K16SwizzleA
using	ck_tile::WarpGemmMfmaF16F16F32M32N32K8TransposedCDistribution
using	ck_tile::WarpGemmMfmaF16F16F32M16N16K16TransposedCDistribution
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaF16F16F32M32N32K16TransposedCDistribution
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaF16F16F32M16N16K32TransposedCDistribution
using	ck_tile::WarpGemmMfmaF16F16F32M32N32K8SwizzleBTransposedCDistribution
using	ck_tile::WarpGemmMfmaF16F16F32M32N32K16SwizzleBTransposedCDistribution
using	ck_tile::WarpGemmMfmaF16F16F32M4N64K16
using	ck_tile::WarpGemmMfmaF16F16F32M64N4K16
using	ck_tile::WarpGemmSmfmacF16F16F32M32N32K16
using	ck_tile::WarpGemmSmfmacF16F16F32M16N16K32
using	ck_tile::WarpGemmMfmaBf16Bf16F32M32N32K8
using	ck_tile::WarpGemmMfmaBf16Bf16F32M16N16K16
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaBf16Bf16F32M32N32K16
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaBf16Bf16F32M16N16K32
using	ck_tile::WarpGemmMfmaBf16Bf16F32M32N32K8SwizzleA
using	ck_tile::WarpGemmMfmaBf16Bf16F32M32N32K16SwizzleA
using	ck_tile::WarpGemmMfmaBf16Bf16F32M32N32K8TransposedCDistribution
using	ck_tile::WarpGemmMfmaBf16Bf16F32M16N16K16TransposedCDistribution
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaBf16Bf16F32M32N32K16TransposedCDistribution
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfmaBf16Bf16F32M16N16K32TransposedCDistribution
using	ck_tile::WarpGemmMfmaBf16Bf16F32M32N32K8SwizzleBTransposedCDistribution
using	ck_tile::WarpGemmMfmaBf16Bf16F32M32N32K16SwizzleBTransposedCDistribution
using	ck_tile::WarpGemmMfmaBf16Bf16F32M4N64K16
using	ck_tile::WarpGemmMfmaBf16Bf16F32M64N4K16
using	ck_tile::WarpGemmMfma_f32_32x32x16_fp8_fp8
using	ck_tile::WarpGemmMfma_f32_32x32x16_fp8_bf8
using	ck_tile::WarpGemmMfma_f32_16x16x32_fp8_bf8
using	ck_tile::WarpGemmMfma_f32_32x32x16_bf8_fp8
using	ck_tile::WarpGemmMfma_f32_32x32x16_bf8_bf8
using	ck_tile::WarpGemmMfma_f32_32x32x32_fp8_fp8
using	ck_tile::WarpGemmMfma_f32_32x32x32_bf8_bf8
using	ck_tile::WarpGemmMfma_f32_32x32x32_fp8_bf8
using	ck_tile::WarpGemmMfma_f32_16x16x32_fp8_fp8
using	ck_tile::WarpGemmMfma_f32_16x16x32_fp8_fp8_CTransposed
using	ck_tile::WarpGemmMfma_f32_16x16x32_bf8_bf8
using	ck_tile::WarpGemmMfma_f32_16x16x32_bf8_bf8_CTransposed
using	ck_tile::WarpGemmMfma_f32_16x16x64_fp8_fp8
using	ck_tile::WarpGemmMfma_f32_16x16x64_bf8_bf8
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_fp4
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_fp8_fp8
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_fp8_bf8
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_bf8_fp8
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_bf8_bf8
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_fp8_fp8_CTransposed
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_fp8_bf8_CTransposed
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_bf8_fp8_CTransposed
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_16x16x128_bf8_bf8_CTransposed
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_32x32x64_fp8_fp8
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_32x32x64_fp8_bf8
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_32x32x64_bf8_fp8
template<WGAttrNumAccessEnum AttrNumAccess = WGAttrNumAccessEnum::Single>
using	ck_tile::WarpGemmMfma_f32_32x32x64_bf8_bf8
using	ck_tile::WarpGemmMfma_f32_32x32x16_fp8_fp8_CTransposed
using	ck_tile::WarpGemmMfma_f32_32x32x16_fp8_bf8_CTransposed
using	ck_tile::WarpGemmMfma_f32_32x32x16_bf8_fp8_CTransposed
using	ck_tile::WarpGemmMfma_f32_32x32x16_bf8_bf8_CTransposed
template<index_t swizzle_factor = 2>
using	ck_tile::WarpGemmMfmaFp8Fp8F32M32N32K32SwizzleBTransposedCDistribution
using	ck_tile::WarpGemmMfma_i32_32x32x16_i8_i8
using	ck_tile::WarpGemmMfma_i32_32x32x16_i8_i8_CTransposed
using	ck_tile::WarpGemmMfma_i32_16x16x32_i8_i8
using	ck_tile::WarpGemmMfma_i32_16x16x32_i8_i8_CTransposed