Aviral Goel bebf0e9d15 Extend Grouped GEMM with MultiD (Single & Double Shared Memory) feature to use persistent kernel option (#2933) ...

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

* 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

* style: clang format

* WIP: host code for grouped_gemm_multi_d persistent kernel compiles but segfaults

* feat(grouped_gemm_multi_d): add functionality to run persistant kernel

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

* 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

* style: clang format

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

* docs: improved README text based on review comments

* fix: parameterize NumDTensor in GroupedGemmHostArgs and remove lint

2025-09-29 15:03:56 -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

Extend Grouped GEMM with MultiD (Single & Double Shared Memory) feature to use persistent kernel option (#2933)

2025-09-29 15:03:56 -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