From 7ec221fb212f5ad99f869e85d7b92a339f99993e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Ville=20Pietil=C3=A4?= <> Date: Fri, 1 Aug 2025 13:35:29 +0000 Subject: [PATCH] Profile only default split-K value. --- .../include/profiler/profile_grouped_conv_bwd_weight_impl.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/profiler/include/profiler/profile_grouped_conv_bwd_weight_impl.hpp b/profiler/include/profiler/profile_grouped_conv_bwd_weight_impl.hpp index 533cd84f8b..5797af0be7 100644 --- a/profiler/include/profiler/profile_grouped_conv_bwd_weight_impl.hpp +++ b/profiler/include/profiler/profile_grouped_conv_bwd_weight_impl.hpp @@ -490,7 +490,7 @@ bool profile_grouped_conv_bwd_weight_impl(int do_verification, range_copy(conv_param.input_left_pads_, begin(input_left_pads)); range_copy(conv_param.input_right_pads_, begin(input_right_pads)); - std::vector fixed_split_k_list = {1, 2, 4, 8, 16, 32, 64, 128}; + std::vector fixed_split_k_list = {1 /*, 2, 4, 8, 16, 32, 64, 128*/}; std::vector best_occupancy_list = {SplitKStrategy::BestOccupancy /*, SplitKStrategy::Optimized*/}; bool profile_all = true; if(split_k != "all")