multi_abd wmma support:

Browse Source

- 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)

...

This commit is contained in:

Enrico Degregori

2025-07-09 12:58:20 +00:00

parent 8f01112ae0

commit 9ee5699e50

37 changed files with 5317 additions and 302 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