[rocm-libraries] ROCm/rocm-libraries#6339 (commit d1ebdab)
feat(composablekernel): [CK][CK_TILE] Add batched contraction multiple ABD (#6339) ## Motivation This PR brings multi-ABD contraction support to CK-Tile, unifying multi-tensor fusion from UniversalGemmKernel / GemmKernelMultiABD with multi-dimensional tensor contraction from BatchedContractionKernel. ISSUE ID : AICK-228 ## Technical Details Added the following: - batched contraction multiple ABD op added to ck_tile - ckTileEngine support - unit-tests - example ## Test Plan Build test_ck_tile_contraction_multi_abd_cshuffle and test_ck_tile_contraction_multi_abd_default2d inside the ROCm Docker container. Run GTest suites which verify GPU kernel output against CPU reference within computed relative/absolute tolerance thresholds. ## Test Result Both test targets compile cleanly with no warnings/errors on gfx942. All test cases pass GPU-vs-CPU verification. ## Submission Checklist - [x] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.
A
aledudek committed
a267ae9a7f9d79eebd1c26d3664894a101398563
Parent: 85ea951
Committed by assistant-librarian[bot] <assistant-librarian[bot]@users.noreply.github.com>
on 7/22/2026, 10:47:39 PM