ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 11:47:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

ck-builder: ck factory grouped conv fwd

Browse Source
This commit is contained in:
Robin Voetter
2025-10-24 14:21:54 +02:00
parent 16db75fadf
commit d15334ed0d
4 changed files with 2357 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
experimental/builder/test/CMakeLists.txt
View File

@@ -31,3 +31,5 @@ function(add_ck_factory_test test_name)
endfunction()
add_ck_factory_test(test_testing_utils test_testing_utils.cpp)
add_ck_factory_test(test_ck_factory_grouped_convolution_forward test_ck_factory_grouped_convolution_forward.cpp)
add_ck_factory_test(test_ck_factory_grouped_convolution_forward_convscale test_ck_factory_grouped_convolution_forward_convscale.cpp)

2345
experimental/builder/test/test_ck_factory_grouped_convolution_forward.cpp Normal file
View File

File diff suppressed because it is too large Load Diff

12
experimental/builder/test/testing_utils.cpp
View File

@@ -252,7 +252,14 @@ bool InstanceMatcher::MatchAndExplain(InstanceSet actual,
*listener << " Missing: " << instances.size() << "\n";
for(const auto& instance : instances)
{
*listener << "- " << instance << "\n";
if(instance == "")
{
*listener << "- (empty string)\n";
}
else
{
*listener << "- " << instance << "\n";
}
}
}
@@ -270,8 +277,7 @@ bool InstanceMatcher::MatchAndExplain(InstanceSet actual,
{
if(instance == "")
{
*listener << "- (empty string; indicates missing GetInstanceString() overload)"
<< "\n";
*listener << "- (empty string)\n";
}
else
{

2
library/include/ck/library/tensor_operation_instance/gpu/grouped_convolution_forward.hpp
View File

@@ -300,7 +300,7 @@ struct DeviceOperationInstanceFactory<ck::tensor_operation::device::DeviceGroupe
#endif
}
// layout NGCHW/GKYXC/NGKHW
// layout NGCHW/GKCYX/NGKHW
if constexpr(NumDimSpatial == 2 && is_same_v<InLayout, NGCHW> &&
is_same_v<WeiLayout, GKCYX> && is_same_v<OutLayout, NGKHW>)
{