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

refactor captures

Browse Source
This commit is contained in:
Max Podkorytov
2025-11-11 21:17:19 -06:00
parent 113d7d65ac
commit 9e6db6fda8
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/ck_tile/03_gemm/universal_gemm_invoker.hpp
View File

@@ -202,9 +202,7 @@ struct UniversalInvoker
const bool has_hot_loop = ck_tile::experimental::builder::HotLoopDescription<CK_TILE_PIPELINE_COMPUTE_V3>::has_hot_loop(num_loop);
const ck_tile::TailNumber tail_num = ck_tile::experimental::builder::HotLoopDescription<CK_TILE_PIPELINE_COMPUTE_V3>::get_tail_num(num_loop);
float ave_time{0};
const auto kernel_launch_visitor = [&](const auto has_hot_loop_,
const auto kernel_launch_visitor = [&args, &s](const auto has_hot_loop_,
const auto tail_number_,
const auto memory_operation_) {
struct Algo
@@ -289,7 +287,7 @@ struct UniversalInvoker
using InputCDEElementWise = CDEElementWise;
};
ave_time = ck_tile::launch_kernel(
float ave_time = ck_tile::launch_kernel(
s, ck_tile::experimental::builder::UniversalFactory<Algo, Inp>::make_kernel(args, s));
return ave_time;
@@ -356,7 +354,7 @@ struct UniversalInvoker
}
};
ave_time = std::visit(
float ave_time = std::visit(
kernel_launch_visitor,
make_bool_variant(has_hot_loop),
make_tailnum_variant(tail_num),