ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-01 20:21:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
7ea1508b59a0e8f89540d8d5f7eb3e7da9a50a62
composable_kernel/include/ck_tile/ops/gemm/kernel
History
Sami Remes 7ea1508b59 [CK_TILE] Move GEMM pipeline tail handling logic to pipelines (#2222) 
* Add TailHandler for V3, V4 and Mem pipelines

* Adapt examples and tests to use TailHandler

* move tail-handling logic to pipeline in persistent grouped gemm

* Fix Mem pipeline dispatching, add CompV4 dispatching

* Use a macro for handling the many tails of Mem pipeline

* Fix formatting again

* Use const-ref RunFunction, remove unnecessary try_run
2025-06-04 11:50:21 +03:00
..
batched_gemm_kernel.hpp
[CK_TILE] Remove scratch usage from universal gemm (#2001)
2025-05-05 18:46:44 +02:00
gemm_kernel.hpp
[CK_TILE] Tile loop persistent gemm kernel (#2191)
2025-06-04 11:46:28 +03:00
gemm_tile_partitioner.hpp
[CK_TILE] Grouped GEMM tile loop (#2146)
2025-05-20 17:18:57 +03:00
grouped_gemm_kernel.hpp
[CK_TILE] Move GEMM pipeline tail handling logic to pipelines (#2222)
2025-06-04 11:50:21 +03:00