ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 13:11:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ce448002ee9988d51b87039fa7769c942f05fd45
composable_kernel/include/ck_tile/ops/gemm/kernel
History
Adam Osewski ce448002ee [CK Tile] Spatially local GEMM tile partitioner. (#1843) 
* Add spatially local tile partitioner

* Use 1D Grid size & create partitioner object.

* Docs & use 1D partitioner in example.

* Clang format.

* Change kernel grid size

Now: X is the # of output C-tiles,
     Y is the batch count
     Z is the splitK

* Formatting & more doc.

* Clang format.

* Fix batched gemm test. Use 1d partitioner.

* Move condition.

* FIx ctor.

* clang-format.
2025-01-31 00:10:16 +01:00
..
batched_gemm_kernel.hpp
[CK Tile] Spatially local GEMM tile partitioner. (#1843)
2025-01-31 00:10:16 +01:00
gemm_kernel.hpp
[CK Tile] Spatially local GEMM tile partitioner. (#1843)
2025-01-31 00:10:16 +01:00
gemm_tile_partitioner.hpp
[CK Tile] Spatially local GEMM tile partitioner. (#1843)
2025-01-31 00:10:16 +01:00
grouped_gemm_kernel.hpp
[CK Tile] Spatially local GEMM tile partitioner. (#1843)
2025-01-31 00:10:16 +01:00