ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-14 19:18:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

correct display info

Browse Source
This commit is contained in:
Lin, Qun
2025-06-07 02:24:30 -05:00
parent 397c9c9865
commit 1f083ab8b5
2 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
example/20_grouped_conv_bwd_weight/device_grouped_conv_bwd_weight_dl_v4.hpp
View File

@@ -1444,12 +1444,13 @@ struct DeviceGroupedConvBwdWeightDlV4 : public DeviceGroupedConvBwdWeight<NDimSp
<< "Dilation<" << Dilation_Y << ", " << Dilation_X << ">, "
<< "Stride<" << Stride_H << ", " << Stride_W<< ">, "
<< "Pad<" << Pad_H << ", " << Pad_W<< ">, "
<< "NBatch: " << NBatch<< ", "
<< "NumWavePerTile: " << NumWavePerTile<< ", "
<< "InScalarPerVector: " << InScalarPerVector<< ", "
<< "OutScalarPerVector: " << OutScalarPerVector<< ", "
<< "DstScalarPerVector: " << DstScalarPerVector<< ", "
<< "RequirePadding: " << RequirePadding << ">"
<< "NBatch: " << NBatch << ", "
<< "NumWavePerTile: " << NumWavePerTile << ", "
<< "InScalarPerVector: " << InScalarPerVector << ", "
<< "OutScalarPerVector: " << OutScalarPerVector << ", "
<< "DstScalarPerVector: " << DstScalarPerVector << ", "
<< "RequirePadding: " << RequirePadding << ", "
<< "WSplit: " << WSplit << ">"
<< std::endl;
// clang-format on

6
example/20_grouped_conv_bwd_weight/run_grouped_conv_bwd_weight_example.inc
View File

@@ -98,9 +98,9 @@ bool run_grouped_conv_bwd_weight(const ExecutionConfig& config,
}
#if ENABLE_CONV_FACTORY
std::size_t best_tflops = 0;
std::size_t best_gb_per_sec = 0;
std::size_t best_avg_time = 0;
float best_tflops = 0;
float best_gb_per_sec = 0;
float best_avg_time = 0;
std::string best_kernel = "";
ck::index_t split_k_array[] = {1, 2, 4, 8, 16, 32};
ck::index_t split_k_count = 6;