include/ck/tensor_operation/gpu/warp/wmma_gemm.hpp File Reference

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