From 9e6db6fda8f5ec06bfa39873eab89c6fa7ce9300 Mon Sep 17 00:00:00 2001 From: Max Podkorytov <4273004+tenpercent@users.noreply.github.com> Date: Tue, 11 Nov 2025 21:17:19 -0600 Subject: [PATCH] refactor captures --- example/ck_tile/03_gemm/universal_gemm_invoker.hpp | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/example/ck_tile/03_gemm/universal_gemm_invoker.hpp b/example/ck_tile/03_gemm/universal_gemm_invoker.hpp index 7de1d268ad..dd38cfe0f3 100644 --- a/example/ck_tile/03_gemm/universal_gemm_invoker.hpp +++ b/example/ck_tile/03_gemm/universal_gemm_invoker.hpp @@ -202,9 +202,7 @@ struct UniversalInvoker const bool has_hot_loop = ck_tile::experimental::builder::HotLoopDescription::has_hot_loop(num_loop); const ck_tile::TailNumber tail_num = ck_tile::experimental::builder::HotLoopDescription::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::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),