ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-02 20:51:23 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4e9b76f88c572a6c54f34cc6467b96279c0e86e4
composable_kernel/example/ck_tile
History
carlushuang 4e9b76f88c [CK_TILE] optimize moe sorting kernel, boost large context case up to 20x (#2153) 
* combine 2-3 as single stage

* support zeroing

* improve long tokens

* update specialization

* b16 ws

* 8bit topk optimize

* update 15 example
2025-05-06 17:32:07 +08:00
..
01_fmha
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
02_layernorm2d
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
03_gemm
[CK_TILE] Remove scratch usage from universal gemm (#2001)
2025-05-05 18:46:44 +02:00
04_img2col
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
05_reduce
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
06_permute
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
09_topk_softmax
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
10_rmsnorm2d
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
11_add_rmsnorm2d_rdquant
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
12_smoothquant
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
13_moe_sorting
[CK_TILE] optimize moe sorting kernel, boost large context case up to 20x (#2153)
2025-05-06 17:32:07 +08:00
14_moe_smoothquant
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
15_fused_moe
[CK_TILE] optimize moe sorting kernel, boost large context case up to 20x (#2153)
2025-05-06 17:32:07 +08:00
16_batched_gemm
[CK_TILE] Remove scratch usage from universal gemm (#2001)
2025-05-05 18:46:44 +02:00
17_grouped_gemm
[CK_TILE] Remove scratch usage from universal gemm (#2001)
2025-05-05 18:46:44 +02:00
18_flatmm
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
35_batched_transpose
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
CMakeLists.txt
Revert "Add ck tile examples to package (#1880)" (#2150)
2025-04-30 10:20:16 -07:00
remod.py
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00