fused_moegemm_traits.hpp File Reference

fused_moegemm_traits.hpp File Reference#

Composable Kernel: fused_moegemm_traits.hpp File Reference
Classes | Namespaces | Enumerations
fused_moegemm_traits.hpp File Reference
#include "ck_tile/core.hpp"

Go to the source code of this file.

Classes

struct  ck_tile::FusedMoeGemmTraits< IsGateOnly_, UseSmoothQuant_, OAtomic_, PermuteEnum_, PadHiddenSize_, PadIntermediateSize_, PipeInterleave_ >

Namespaces

namespace  ck_tile

Enumerations

enum class  ck_tile::FusedMoeGemmWeightPermuteEnum {
  ck_tile::no_permute = 0 ,
  ck_tile::b_nr_kr_kw_nw_kv = 1 ,
  ck_tile::b_nr_kr_waveflatten = b_nr_kr_kw_nw_kv
}
enum class  ck_tile::FusedMoeGemmPipelineSequencerEnum {
  ck_tile::SLD_A = 1 << 0 ,
  ck_tile::SLD_B = 1 << 1 ,
  ck_tile::GLD_A = 1 << 2 ,
  ck_tile::GLD_B = 1 << 3 ,
  ck_tile::SST_A = 1 << 4 ,
  ck_tile::SST_B = 1 << 5 ,
  ck_tile::GST_O = 1 << 6
}