fused_moe.hpp Source File# Composable Kernel: fused_moe.hpp Source File includeck_tileops fused_moe.hpp Go to the documentation of this file. 1// SPDX-License-Identifier: MIT 2// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved. 3 4#pragma once 5 6#include "ck_tile/ops/fused_moe/kernel/fused_moegemm_kernel.hpp" 7#include "ck_tile/ops/fused_moe/kernel/fused_moegemm_shape.hpp" 8#include "ck_tile/ops/fused_moe/kernel/fused_moegemm_tile_partitioner.hpp" 9#include "ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp" 10#include "ck_tile/ops/fused_moe/kernel/moe_sorting_problem.hpp" 11#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_ex.hpp" 12#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_policy.hpp" 13#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp" 14#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_problem.hpp" 15#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_traits.hpp" 16#include "ck_tile/ops/fused_moe/pipeline/moe_sorting_pipeline.hpp" 17#include "ck_tile/ops/fused_moe/pipeline/moe_sorting_policy.hpp" 18#include "ck_tile/ops/common/generic_2d_block_shape.hpp" 19#include "ck_tile/ops/common/load_interleaved_pk_type.hpp" 20#include "ck_tile/ops/common/streamk_common.hpp" 21#include "ck_tile/ops/common/tensor_layout.hpp" 22#include "ck_tile/ops/common/utils.hpp" fused_moegemm_kernel.hpp fused_moegemm_pipeline_flatmm_ex.hpp fused_moegemm_pipeline_flatmm_policy.hpp fused_moegemm_pipeline_flatmm_uk.hpp fused_moegemm_pipeline_problem.hpp fused_moegemm_shape.hpp fused_moegemm_tile_partitioner.hpp fused_moegemm_traits.hpp generic_2d_block_shape.hpp load_interleaved_pk_type.hpp moe_sorting_kernel.hpp moe_sorting_pipeline.hpp moe_sorting_policy.hpp moe_sorting_problem.hpp streamk_common.hpp tensor_layout.hpp utils.hpp