/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/warp/wmma_gemm.hpp File Reference

• develop
• include
• ck
• tensor_operation
• gpu
• warp

Classes | Namespaces | Enumerations

wmma_gemm.hpp File Reference

#include "ck/utility/common_header.hpp"
#include "ck/utility/math.hpp"
#include "ck/utility/amd_wmma.hpp"

Go to the source code of this file.

Classes
struct	ck::wmma_type< Instr, WaveSize, typename >
struct	ck::wmma_type< WmmaInstr::wmma_f32_16x16x16_f16, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_f32_16x16x16_bf16, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_f16_16x16x16_f16, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_bf16_16x16x16_bf16, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_i32_16x16x16_iu8, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_f32_16x16x16_f16_gfx12, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_f32_16x16x16_bf16_gfx12, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_i32_16x16x16_iu8_gfx12, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_f32_16x16x16_f8f8_gfx12, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_f32_16x16x16_f8bf8_gfx12, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_f32_16x16x16_bf8f8_gfx12, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::wmma_type< WmmaInstr::wmma_f32_16x16x16_bf8bf8_gfx12, WaveSize, typename std::enable_if_t< WaveSize==32||WaveSize==64 > >
struct	ck::WmmaSelector< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma >
struct	ck::WmmaGemm< src_type_a, src_type_b, dst_type, MPerWmma, NPerWmma, KPack, TransposeC, AssemblyBackend >

Namespaces
	ck

Enumerations
enum class	ck::WmmaInstr {   ck::wmma_f32_16x16x16_f16 = 0 ,   ck::wmma_f32_16x16x16_bf16 ,   ck::wmma_f16_16x16x16_f16 ,   ck::wmma_bf16_16x16x16_bf16 ,   ck::wmma_i32_16x16x16_iu8 ,   ck::wmma_i32_16x16x16_iu4 ,   ck::wmma_f32_16x16x16_f16_gfx12 ,   ck::wmma_f32_16x16x16_bf16_gfx12 ,   ck::wmma_i32_16x16x16_iu8_gfx12 ,   ck::wmma_f32_16x16x16_f8f8_gfx12 ,   ck::wmma_f32_16x16x16_f8bf8_gfx12 ,   ck::wmma_f32_16x16x16_bf8f8_gfx12 ,   ck::wmma_f32_16x16x16_bf8bf8_gfx12 }