mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-16 10:59:55 +00:00
8acce2dee1
* 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>
[ROCm/composable_kernel commit: 440e28b08f]
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"
|