ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-13 10:37:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

Profile only default split-K value.

Browse Source
This commit is contained in:
Ville Pietilä
2025-08-01 13:35:29 +00:00
parent f766ff7bbf
commit 7ec221fb21
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
profiler/include/profiler/profile_grouped_conv_bwd_weight_impl.hpp
View File

@@ -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<ck::index_t> fixed_split_k_list = {1, 2, 4, 8, 16, 32, 64, 128};
std::vector<ck::index_t> fixed_split_k_list = {1 /*, 2, 4, 8, 16, 32, 64, 128*/};
std::vector<SplitKStrategy> best_occupancy_list = {SplitKStrategy::BestOccupancy /*, SplitKStrategy::Optimized*/};
bool profile_all = true;
if(split_k != "all")