ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-04 21:51:28 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
3db77bc4f26453a5ba5aad3d49adb03d1accf8de
composable_kernel/example/ck_tile
History
Mateusz Ozga 3c93d3c444 CK-Tile Grouped GEMM refactor and post PR fixes (#1756) 
* Grouped gemm simple code refactor

* Offset invoker

* Invoke generic Run, and replace name of parrtitioner variable

* Tests fix type

* Removed namespaces

* Add template param to avoid implicit cast

* Remove generic function

* Constant value

* underline enum to int16_t

* Generalize partitioner function

* Remove whitespaces

* Rename function

* Using support

* Clang-format

* Clang-format

* Fn-partitioner description fn

* Typo

* Typo 2

* Better description

* Better description

* Refactor after review

* Use ctr instead of set fn

* Inovke ctr and typo

* Comments

* Remove unnecessary comment

* Review, remove modulo
2025-01-21 21:06:10 +01:00
..
01_fmha
[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)
2025-01-07 18:49:24 +08:00
02_layernorm2d
[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)
2025-01-15 10:23:48 +08:00
03_gemm
CK-Tile Grouped GEMM refactor and post PR fixes (#1756)
2025-01-21 21:06:10 +01:00
04_img2col
[CK_TILE] Update example README files & fix script compatibility issue (#1548)
2024-10-08 10:45:12 +08:00
05_reduce
[CK_TILE] Adjust kBlockSize of reduce example for better perf (#1779)
2025-01-12 20:50:32 -08:00
06_permute
[CK_TILE] fused-moe first version (#1634)
2024-11-26 11:14:56 +08:00
09_topk_softmax
topk_softmax (#1592)
2024-10-26 23:52:49 +08:00
10_rmsnorm2d
[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)
2025-01-15 10:23:48 +08:00
11_add_rmsnorm2d_rdquant
[Ck_tile] smoothquant (#1617)
2024-11-01 13:51:56 +08:00
12_smoothquant
[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)
2025-01-15 10:23:48 +08:00
13_moe_sorting
[CK_TILE] optimize moe-sorting kernel (#1771)
2024-12-23 10:59:02 +08:00
14_moe_smoothquant
[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)
2025-01-15 10:23:48 +08:00
15_fused_moe
[CK_TILE] Fix mock token id, support g1u1/g1u0 through same inline code block (#1808)
2025-01-16 17:51:10 +08:00
16_batched_gemm
CK-Tile Grouped GEMM refactor and post PR fixes (#1756)
2025-01-21 21:06:10 +01:00
17_grouped_gemm
CK-Tile Grouped GEMM refactor and post PR fixes (#1756)
2025-01-21 21:06:10 +01:00
CMakeLists.txt
Ck tile grouped GEMM example (#1713)
2024-12-04 21:40:01 +01:00
remod.py
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00