Aviral Goel a44bea45b2 Integrate Multi D GEMMs into Grouped GEMMs along with unit tests (#2923) ...

* feat(grouped_gemm_multi_d): add new example that integrates grouped_gemm and multi_d_gemm feature

* feat: generalized grouped_gemm_kernel.hpp

* feat: generalized grouped_gemm_kernel.hpp even further by removing hardcoded 0

* refactor: grouped_gemm_multi_d relies on grouped_gemm_kernel

* tests(grouped_gemm): grouped_gemm test suite passes with minor adjustments

* fix: segfault fix by passing correct parameters for d tensors

* docs: add multi d info and trim down outdated content

* tests: add unit tests for grouped_gemm_multi_d and minor changes in grouped_gemm related test for compatibility

* style: clang format

* fix: incorrect validation method and Dtensor layout in test suite

2025-09-26 09:59:58 -07:00

..

batched_gemm_kernel.hpp

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -07:00

gemm_kernel.hpp

[CK_TILE] Multiple-ABD GEMM example (#2788)

2025-09-19 01:14:11 +02:00

gemm_multi_abd_kernel.hpp

Fix the gfx950 numerical errors (#2911)

2025-09-23 22:54:52 -07:00

gemm_multi_d_kernel.hpp

[CK_TILE] Multiple-ABD GEMM example (#2788)

2025-09-19 01:14:11 +02:00

gemm_tile_partitioner.hpp

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -07:00

grouped_gemm_kernel.hpp

Integrate Multi D GEMMs into Grouped GEMMs along with unit tests (#2923)

2025-09-26 09:59:58 -07:00

streamk_gemm_kernel.hpp

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -07:00

universal_gemm_kernel.hpp

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -07:00