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

Fix 8 waves operator to use Problem defined dispatching for hot loop and tail

Browse Source
This commit is contained in:
Enrico Degregori
2026-07-01 13:04:38 +00:00
parent 8207ddfcd1
commit 0523204792
1 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
include/ck_tile/ops/gemm/pipeline/gemm_pipeline_ag_bg_cr_comp_async_eight_waves.hpp
View File

@@ -248,12 +248,13 @@ struct GemmPipelineAgBgCrCompAsyncEightWaves : public BaseGemmPipelineAgBgCrComp
index_t num_loop,
void* p_smem) const
{
return operator()(ck_tile::make_tuple(a_dram_block_window_tmp),
ck_tile::element_wise::PassThrough{},
ck_tile::make_tuple(b_dram_block_window_tmp),
ck_tile::element_wise::PassThrough{},
num_loop,
p_smem);
return PipelineImpl<Scheduler>{}.template operator()<Problem::HasHotLoop, Problem::TailNum>(
a_dram_block_window_tmp,
ck_tile::element_wise::PassThrough{},
b_dram_block_window_tmp,
ck_tile::element_wise::PassThrough{},
num_loop,
p_smem);
}
};