ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 08:50:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add complete multi-dimensional stride support via descriptors

Browse Source
This commit is contained in:
Mohsen Saffari
2025-10-20 14:43:32 +00:00
parent b8b56d5cc6
commit bbfe4501fa
2 changed files with 64 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
example/ck_tile/41_batched_contraction/batched_contraction.cpp
View File

@@ -219,9 +219,8 @@ float batched_contraction(const ck_tile::BatchedContractionHostArgs<DsDataType::
HANDLE_CASE(2, 1, 1, 1);
HANDLE_CASE(2, 2, 2, 1);
HANDLE_CASE(1, 2, 1, 1);
HANDLE_CASE(1, 1, 1, 2);
HANDLE_CASE(2, 1, 1, 1);
HANDLE_CASE(2, 2, 2, 2);
HANDLE_CASE(4, 4, 4, 4);
throw std::runtime_error(
"Unsupported dimension combination: G=" + std::to_string(num_g_dims) +