mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-03 05:01:25 +00:00
440e28b08f
* moe pipeline * update code * compile OK * update * update cpu reference * update pipeline_gemm0 * compiler ok * update pipeline * rename to ex pipeline * block-asm * update * update * update first gemm ok * compute correct * update file structure * update README * update * update * update code * update API * return unsupport case * add comment * update readme * update * uncomment * update * fix build err --------- Co-authored-by: valarLip <340077269@qq.com>
20 lines
1.1 KiB
C++
20 lines
1.1 KiB
C++
// SPDX-License-Identifier: MIT
|
|
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
#pragma once
|
|
|
|
#include "ck_tile/ops/fused_moe/kernel/fused_moegemm_kernel.hpp"
|
|
#include "ck_tile/ops/fused_moe/kernel/fused_moegemm_shape.hpp"
|
|
#include "ck_tile/ops/fused_moe/kernel/fused_moegemm_tile_partitioner.hpp"
|
|
#include "ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp"
|
|
#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_ex.hpp"
|
|
#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_policy.hpp"
|
|
#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_flatmm_uk.hpp"
|
|
#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_problem.hpp"
|
|
#include "ck_tile/ops/fused_moe/pipeline/fused_moegemm_traits.hpp"
|
|
#include "ck_tile/ops/fused_moe/pipeline/moe_sorting_pipeline.hpp"
|
|
#include "ck_tile/ops/fused_moe/pipeline/moe_sorting_policy.hpp"
|
|
#include "ck_tile/ops/fused_moe/pipeline/moe_sorting_problem.hpp"
|
|
#include "ck_tile/ops/common/generic_2d_block_shape.hpp"
|
|
#include "ck_tile/ops/common/tensor_layout.hpp"
|