Erwin Terpstra fe40a5d139 Implement batched gemm bias permute for RDNA4 (#3534) ...

* feat: test setup for batched contraction (aka batched gemm multiple d e permute)

* wip: device struct for WMMA batched contraction multiple d based on new gridwise op

* feat: working batched contraction on RDNA, non-naive tensor descriptors for gridwise_gemm_wmma_cshuffle_v3, test setup for odd cases

* fix: failure to resolve template parameters when calling new function overload

* fix: passing reference type as parameter instead of underlying types

* fix: merge error caused duplicate definitions

* fix: make sure constness of template and parameters types match

* fix: don't compile batched contraction test on unsupported architectures

* feat: add example for new wmma implementation, and consolidate example code between platforms

* style: return inline instead of with branch

* chore: add extra assert on vector memory access sizes

* chore: clean up some unused variables

* fix: correct tail number calculation, added small cases and extra instances to the test

* fix: properly support wave transfer by generating correct grid descriptors dependent on the transfer method

2026-01-17 08:30:27 +01:00

..

CMakeLists.txt

Implement batched gemm bias permute for RDNA4 (#3534)

2026-01-17 08:30:27 +01:00

test_batched_contraction.cpp

Implement batched gemm bias permute for RDNA4 (#3534)

2026-01-17 08:30:27 +01:00