wmma_gemm.hpp File Reference

wmma_gemm.hpp File Reference#

Composable Kernel: wmma_gemm.hpp File Reference
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

namespace  ck

Enumerations

enum struct  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
}