ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-01 04:07:56 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add more instances.

Browse Source
This commit is contained in:
Ville Pietilä
2025-10-03 14:39:31 +00:00
parent 48d22d2b9b
commit 970f037606
1 changed files with 21 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
example/ck_tile/20_grouped_convolution/grouped_convolution_backward_weight.cpp
View File

@@ -218,9 +218,29 @@ int run(const std::string& in_layout,
{
return run_grouped_conv_bwd_weight_example_prec_type<InPrecType>(in_layout, wei_layout, out_layout, argc, argv);
}
else if (num_groups_to_merge == 2)
{
return run_grouped_conv_bwd_weight_example_prec_type<InPrecType, 2>(in_layout, wei_layout, out_layout, argc, argv);
}
else if (num_groups_to_merge == 4)
{
return run_grouped_conv_bwd_weight_example_prec_type<InPrecType, 4>(in_layout, wei_layout, out_layout, argc, argv);
}
else if (num_groups_to_merge == 8)
{
return run_grouped_conv_bwd_weight_example_prec_type<InPrecType>(in_layout, wei_layout, out_layout, argc, argv);
return run_grouped_conv_bwd_weight_example_prec_type<InPrecType, 8>(in_layout, wei_layout, out_layout, argc, argv);
}
else if (num_groups_to_merge == 16)
{
return run_grouped_conv_bwd_weight_example_prec_type<InPrecType, 16>(in_layout, wei_layout, out_layout, argc, argv);
}
else if (num_groups_to_merge == 32)
{
return run_grouped_conv_bwd_weight_example_prec_type<InPrecType, 32>(in_layout, wei_layout, out_layout, argc, argv);
}
else if (num_groups_to_merge == 64)
{
return run_grouped_conv_bwd_weight_example_prec_type<InPrecType, 64>(in_layout, wei_layout, out_layout, argc, argv);
}
else
{