Skip to content

GitLab

"host/host_tensor/include/device.hpp" did not exist on "c15ff3c825dd25ddc2f6ee572f867c925a4528bd"
Switch branch/tag
Find file Normal viewHistoryPermalink
flatmm.hpp 596 Bytes
Newer Older
carlushuang's avatar
[CK_TILE] fused-moe first version (#1634)  
carlushuang committed
1
2
3
4
5
6
7 
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.

#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"
coderfeli's avatar
build ok  
coderfeli committed
9 
#include "ck_tile/ops/flatmm/block/flatmm_sn_32x128x256_1x4x1_16x16x32_itl.hpp"
carlushuang's avatar
[CK_TILE] fused-moe first version (#1634)  
carlushuang committed
10
11
12 
#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"