ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-12 09:16:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP: v4 tile distribution working

Browse Source
This commit is contained in:
Damien Lejeune
2026-02-10 13:55:07 +00:00
parent 7c728adb57
commit 63dcefffc3
6 changed files with 413 additions and 125 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
example/ck_tile/42_mhc/mhc_v4_bf16_benchmark.cpp
View File

@@ -94,13 +94,8 @@ bool run_mhc_benchmark(const ck_tile::ArgParser& arg_parser)
d_phi_mem.ToDevice(h_phi.data());
d_output_mem.ToDevice(h_output.data());
// Define block shape - 64 threads (1 warp) to match BlockGemmShape configuration
// This matches a 16x16 block tile with 1 warp (1x1 warp layout)
using BlockShape = ck_tile::Generic2dBlockShape<ck_tile::sequence<1, 64>,
ck_tile::sequence<1, 64>,
ck_tile::sequence<1, 1>>;
using Problem = ck_tile::MHCProblem<XDataType, ComputeDataType, YDataType, BlockShape>;
// Use MHCProblemV4 which automatically derives BlockShape from BlockGemmShape
using Problem = ck_tile::MHCProblemV4<XDataType, ComputeDataType, YDataType>;
// V4 kernel - optimized with single-pass data loading
using KernelV4 = ck_tile::MHCKernelV4<Problem, ck_tile::MHCDefaultPolicy, ActivationFunc>;