Wmma support for multiple ABD GEMM (#2803)

Browse Source

* multi_abd wmma support:

 - Add multiple A and B support to multiple D implementation (gridwise level)
 - Add multi_abd GEMM (device level)
 - Add instances (xdl parity)
 - Add tests (both xdl and wmma)
 - Add examples
 - Add ckProfiler support (both xdl and wmma)

* Fix bug in device print function

* Fix unused template parameter

* Fix batched gemm for multiABD gridwise implementation

* Fix gemm_universal_reduce with multiABDs gridwise implementation

---------

Co-authored-by: Illia Silin <98187287+illsilin@users.noreply.github.com>

[ROCm/composable_kernel commit: 3d29bff2f0]

...

This commit is contained in:

Enrico Degregori

2025-09-23 03:49:06 +02:00

committed by GitHub

parent 7122e27fd8

commit 0b149c8695

38 changed files with 5343 additions and 312 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

954

library/include/ck/library/tensor_operation_instance/gpu/gemm_multi_abd.hpp

File diff suppressed because it is too large Load Diff