/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_problem.hpp Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_problem.hpp Source File#

Composable Kernel: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck_tile/ops/fused_moe/pipeline/fused_moegemm_pipeline_problem.hpp Source File
fused_moegemm_pipeline_problem.hpp
Go to the documentation of this file.
1 // SPDX-License-Identifier: MIT
2 // Copyright (c) 2024, Advanced Micro Devices, Inc. All rights reserved.
3 
4 #pragma once
5 
6 #include "ck_tile/core.hpp"
7 
8 namespace ck_tile {
9 
10 // TODO: alow 2 gemm have different type
11 template <typename ADataType_,
12  typename GDataType_,
13  typename DDataType_,
14  typename AccDataType_,
15  typename ODataType_,
16  typename AScaleDataType_,
17  typename GScaleDataType_,
18  typename DScaleDataType_,
19  typename YSmoothScaleDataType_,
20  typename TopkWeightDataType_,
21  typename IndexDataType_, // data type for all indexing
22  typename GateActivation_, // = ck_tile::element_wise::Silu,
23  typename BlockShape_, // shoule be FusedMoeGemmShape
24  typename Traits_>
26 {
38 
39  // the input for next gemm should have same time as
41 
45 };
46 } // namespace ck_tile
Definition: cluster_descriptor.hpp:13
remove_cv_t< std::remove_reference_t< T > > remove_cvref_t
Definition: type_traits.hpp:21
Definition: fused_moegemm_pipeline_problem.hpp:26
remove_cvref_t< ADataType_ > ADataType
Definition: fused_moegemm_pipeline_problem.hpp:27
remove_cvref_t< BlockShape_ > BlockShape
Definition: fused_moegemm_pipeline_problem.hpp:43
remove_cvref_t< GScaleDataType_ > GScaleDataType
Definition: fused_moegemm_pipeline_problem.hpp:33
remove_cvref_t< AccDataType_ > AccDataType
Definition: fused_moegemm_pipeline_problem.hpp:30
remove_cvref_t< IndexDataType_ > IndexDataType
Definition: fused_moegemm_pipeline_problem.hpp:37
remove_cvref_t< Traits_ > Traits
Definition: fused_moegemm_pipeline_problem.hpp:44
remove_cvref_t< TopkWeightDataType_ > TopkWeightDataType
Definition: fused_moegemm_pipeline_problem.hpp:36
remove_cvref_t< DDataType_ > DDataType
Definition: fused_moegemm_pipeline_problem.hpp:29
remove_cvref_t< AScaleDataType_ > AScaleDataType
Definition: fused_moegemm_pipeline_problem.hpp:32
ADataType YDataType
Definition: fused_moegemm_pipeline_problem.hpp:40
remove_cvref_t< GDataType_ > GDataType
Definition: fused_moegemm_pipeline_problem.hpp:28
remove_cvref_t< YSmoothScaleDataType_ > YSmoothScaleDataType
Definition: fused_moegemm_pipeline_problem.hpp:35
remove_cvref_t< ODataType_ > ODataType
Definition: fused_moegemm_pipeline_problem.hpp:31
remove_cvref_t< DScaleDataType_ > DScaleDataType
Definition: fused_moegemm_pipeline_problem.hpp:34
remove_cvref_t< GateActivation_ > GateActivation
Definition: fused_moegemm_pipeline_problem.hpp:42