ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-04-20 14:59:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[CK_BUILDER] Add grouped conv fwd ck tile profiler (#3518)

Browse Source 
* [BULDER] Add grouped conv fwd ck tile profiler

* [CK TILE] Fix grouped conv kernels splitk and double lds

* Updates

* Fixes

* Move to ckProfiler

* Fixes

* fix

* fix

* Change instances to empty list by default

* fix

* fix

* Update grouped_convolution_signatures.hpp

* Update grouped_convolution_forward_tile_algs.hpp

* [CK TILE] Add grouped convolution forward tests (#3556)

* [CK TILE] Add grouped convolution forward tests

* fix jenkins

* fixes

* comments fixes

* unit test

* unit test fix

* Move instances outside builder

* fix includes

* clang format fix

* readme fix

* fix includes

* fixes
This commit is contained in:
Bartłomiej Kocot
2026-01-20 06:29:01 +01:00
committed by GitHub
parent 0517d43d31
commit 0727e85e52
44 changed files with 3083 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
include/ck/library/utility/host_tensor.hpp
View File

@@ -298,9 +298,12 @@ struct HostTensorDescriptor
if constexpr(!(std::is_same_v<ck::tensor_layout::gemm::RowMajor, Layout> ||
std::is_same_v<ck::tensor_layout::gemm::ColumnMajor, Layout>))
{
std::cerr << "Only RowMajor and ColumnMajor layouts are supported for empty "
"strides, got "
<< layout << ". Will calculate strides as RowMajor." << std::endl;
if(dbg)
{
std::cerr << "Only RowMajor and ColumnMajor layouts are supported for empty "
"strides, got "
<< layout << ". Will calculate strides as RowMajor." << std::endl;
}
}
mStrides.clear();
@@ -443,9 +446,14 @@ struct HostTensorDescriptor
{
// TBD: implement verification for Conv layouts
// For now, just print warning and return
std::cerr << "Warning: Tensor layout verification for ck::tensor_layout::convolution "
"layouts is not supported yet. Skipping..."
<< std::endl;
if(dbg)
{
std::cerr
<< "Warning: Tensor layout verification for ck::tensor_layout::convolution "
"layouts is not supported yet. Skipping..."
<< std::endl;
}
return;
}
else