Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
flatmm.hpp 516 Bytes
Newer Older
carlushuang's avatar
[CK_TILE] fused-moe first version (#1634)  
carlushuang committed
1 
// SPDX-License-Identifier: MIT
Po Yen Chen's avatar
[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)  
Po Yen Chen committed
2 
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.
carlushuang's avatar
[CK_TILE] fused-moe first version (#1634)  
carlushuang committed
3
4
5
6
7 

#pragma once

#include "ck_tile/ops/flatmm/block/flatmm_32x512x128_1x4x1_16x16x32.hpp"
#include "ck_tile/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32.hpp"
Xu, Shengnan's avatar
added moe interleaving pipeline (#1712)  
Xu, Shengnan committed
8 
#include "ck_tile/ops/flatmm/block/flatmm_sn_32x128x512_1x4x1_16x16x32_itl.hpp"
carlushuang's avatar
[CK_TILE] fused-moe first version (#1634)  
carlushuang committed
9
10
11 
#include "ck_tile/ops/flatmm/block/flatmm_uk_config.hpp"
#include "ck_tile/ops/common/generic_2d_block_shape.hpp"
#include "ck_tile/ops/common/tensor_layout.hpp"