zanzhang b324bbe45f Merge branch 'develop' into zan_vllm

2025-05-14 13:11:55 +08:00

..

01_fmha

Merge branch 'develop' into zan_vllm

2025-05-14 13:11:55 +08:00

02_layernorm2d

[CK_TILE] Improve RMS/Layer Normalization 2 Pass Pipeline Performance (#1861)

2025-03-25 20:09:45 +08:00

03_gemm

[CK_Tile] Simplified Mem pipeline (#2159)

2025-05-07 18:37:31 -07: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] Improve RMS/Layer Normalization 2 Pass Pipeline Performance (#1861)

2025-03-25 20:09:45 +08:00

11_add_rmsnorm2d_rdquant

Revert "Add ck tile examples to package (#1880)" (#2150)

2025-04-30 10:20:16 -07: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, boost large context case up to 20x (#2153)

2025-05-06 17:32:07 +08:00

14_moe_smoothquant

add fp8 as dst (#1830)

2025-01-22 17:34:27 +08: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

Update the buffer load/store intrinsic names for clang>=20. (#2192)

2025-05-13 10:18:14 -07:00

35_batched_transpose

Vectorized Transpose for Batched Transpose CK Tile Operator (#2131)

2025-05-12 00:41:45 -07:00

36_copy

Simple copy kernel, which can be a tool to experiment with CK_Tile API with minimal code. (#2156)

2025-05-07 00:02:59 -07:00

CMakeLists.txt

Simple copy kernel, which can be a tool to experiment with CK_Tile API with minimal code. (#2156)

2025-05-07 00:02:59 -07:00

remod.py

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00