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

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

Classes | Namespaces | Typedefs | Variables

warp_gemm_attribute_wmma_impl.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/core/config.hpp"

Go to the source code of this file.

Classes
struct	ck_tile::WarpGemmAttributeWmmaImpl< Traits >
struct	ck_tile::has_wmma_traits< Arch, AType, BType, CType, warp_m, warp_n, warp_k >

Namespaces
	ck_tile

Typedefs
using	ck_tile::DeviceIp = remove_cvref_t< decltype(ck_tile::get_device_arch())>
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_f16_f16 = WarpGemmAttributeWmmaImpl< WmmaTraits< DeviceIp, fp16_t, fp16_t, float, 16, 16, 16 > >
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_bf16_bf16 = WarpGemmAttributeWmmaImpl< WmmaTraits< DeviceIp, bf16_t, bf16_t, float, 16, 16, 16 > >
using	ck_tile::WarpGemmAttributeWmmaImpl_i32_16x16x16_i8_i8 = WarpGemmAttributeWmmaImpl< WmmaTraits< DeviceIp, int8_t, int8_t, int32_t, 16, 16, 16 > >
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_f8 = WarpGemmAttributeWmmaImpl< WmmaTraits< gfx12_t, fp8_t, fp8_t, float, 16, 16, 16 > >
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_bf8 = WarpGemmAttributeWmmaImpl< WmmaTraits< gfx12_t, bf8_t, bf8_t, float, 16, 16, 16 > >
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_f8_bf8 = WarpGemmAttributeWmmaImpl< WmmaTraits< gfx12_t, fp8_t, bf8_t, float, 16, 16, 16 > >
using	ck_tile::WarpGemmAttributeWmmaImpl_f32_16x16x16_bf8_f8 = WarpGemmAttributeWmmaImpl< WmmaTraits< gfx12_t, bf8_t, fp8_t, float, 16, 16, 16 > >

Variables
template<typename Arch , typename AType , typename BType , typename CType , index_t warp_m, index_t warp_n, index_t warp_k>
constexpr bool	ck_tile::has_wmma_traits_v