mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-20 04:49:54 +00:00
326639e80c
* Generate group mode paged-attn kernel
* Enable paged-kvcache + group mode support
* Add missing header: fused_moe.hpp
* Add comment to explain kernel arg usage
* Make error message more clear
* Add comment for confusing data member names
* Add more comment for confusing variable names
* Fix typo in option description
[ROCm/composable_kernel commit: fb1ccfa9df]
12 lines
491 B
C++
12 lines
491 B
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/moe_sorting_kernel.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"
|