/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/flatmm/kernel/moe_flatmm_kernel.hpp File Reference

• develop
• include
• ck_tile
• ops
• flatmm
• kernel

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
	ck_tile
	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 }