moe_flatmm_kernel.hpp File Reference

moe_flatmm_kernel.hpp File Reference#

Composable Kernel: moe_flatmm_kernel.hpp File Reference
Classes | Namespaces | Enumerations
moe_flatmm_kernel.hpp File Reference
#include "ck_tile/core/numeric/math.hpp"
#include "ck_tile/core/utility/literals.hpp"
#include "ck_tile/ops/elementwise/unary_element_wise_operation.hpp"
#include "ck_tile/ops/flatmm/kernel/flatmm_kernel.hpp"
#include "ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp"
#include "ck_tile/host.hpp"

Go to the source code of this file.

Classes

struct  ck_tile::MoeFlatmmHostArgs< ScaleM, ScaleN, ExpertBias >
struct  ck_tile::moe::MoeSilu
struct  ck_tile::moe::Swiglu
struct  ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >
struct  ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >::MoeFlatmmKernelArgs< ScaleM, ScaleN, ExpertBias >
struct  ck_tile::MoeFlatmmKernel< TilePartitioner_, FlatmmPipeline_, EpiloguePipeline_, kind, FusedActivation >::SplitKBatchOffset

Namespaces

namespace  ck_tile
namespace  ck_tile::moe

Enumerations

enum class  ck_tile::MoeFlatmmKind {
  ck_tile::kFFN_gemm1_gate_only ,
  ck_tile::kFFN_gemm1_gate_up ,
  ck_tile::kFFN_gemm2
}