ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-08 08:07:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

port ck DeviceOpInstance to ck tile

Browse Source
This commit is contained in:
zufayu
2025-08-13 22:14:50 +08:00
parent 1698930818
commit 2f04e78d96
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/ck_tile/39_gemm_microscaling/gemm_mx_fp4_basic.cpp
View File

@@ -35,9 +35,9 @@ float gemm_mx_calc(const ck_tile::GemmMXHostArgs& args, const ck_tile::stream_co
static_assert(std::is_same_v<CLayout, ck_tile::tensor_layout::gemm::RowMajor>);
constexpr ck_tile::index_t M_Tile = 64;
constexpr ck_tile::index_t N_Tile = 64;
constexpr ck_tile::index_t K_Tile = 128;
constexpr ck_tile::index_t M_Tile = 256;
constexpr ck_tile::index_t N_Tile = 256;
constexpr ck_tile::index_t K_Tile = 128; // 256 / DataPackedSize (Packed representation of data)
constexpr ck_tile::index_t M_Warp = 2;
constexpr ck_tile::index_t N_Warp = 2;
@@ -113,7 +113,7 @@ float gemm_mx_calc(const ck_tile::GemmMXHostArgs& args, const ck_tile::stream_co
M_Warp_Tile,
N_Warp_Tile,
K_Warp_Tile,
y transposed_warp_gemm,
transposed_warp_gemm,
ck_tile::memory_operation_enum::set>>;
using Kernel = ck_tile::GemmMXKernel<TilePartitioner, CodegenGemmPipeline, GemmEpilogue>;