include/ck_tile/ops/fused_moe.hpp Source File# Composable Kernel: include/ck_tile/ops/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/pipeline/fused_moegemm_pipeline_flatmm_ex.hpp" 11 #include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_policy.hpp" 12 #include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp" 13 #include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_problem.hpp" 14 #include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_traits.hpp" 15 #include "ck_tile/ops/fused_moe/pipeline/moe_sorting_pipeline.hpp" 16 #include "ck_tile/ops/fused_moe/pipeline/moe_sorting_policy.hpp" 17 #include "ck_tile/ops/fused_moe/pipeline/moe_sorting_problem.hpp" 18 #include "ck_tile/ops/common/generic_2d_block_shape.hpp" 19 #include "ck_tile/ops/common/tensor_layout.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 moe_sorting_kernel.hpp moe_sorting_pipeline.hpp moe_sorting_policy.hpp moe_sorting_problem.hpp tensor_layout.hpp