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