Clement Lin 399cdb6f9f Adjust the size of thread block ...

1968.42 GB/s -> 1998.24 GB/s

2025-03-12 14:33:36 +08:00

..

01_fmha

refactor ck-tile kernel launch (#1925)

2025-03-07 08:29:40 -08:00

02_layernorm2d

Implement fp8 quant for layernorm and rmsnorm (#1814)

2025-01-24 16:40:43 +08:00

03_gemm

Addressing (Post Merge) code review comments for PR 1845 (#1883)

2025-03-06 11:40:30 -08: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

Implement fp8 quant for layernorm and rmsnorm (#1814)

2025-01-24 16:40:43 +08:00

11_add_rmsnorm2d_rdquant

[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)

2025-02-20 14:00:27 -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] add moe-sorting MP kernel (#1910)

2025-02-25 17:56:55 +08:00

14_moe_smoothquant

add fp8 as dst (#1830)

2025-01-22 17:34:27 +08:00

15_fused_moe

[CK_TILE] add moe-sorting MP kernel (#1910)

2025-02-25 17:56:55 +08:00

16_batched_gemm

[CK_TILE] Add GetName for GEMM kernels (#1791)

2025-02-11 09:49:48 +01:00

17_grouped_gemm

[CK TILE] Gemm pk_int4_t permute B (#1907)

2025-02-27 11:01:14 +01:00

35_batched_transpose

add batched_transpose implement (#1660)

2025-01-28 16:22:02 -08:00

99_toy_example

Adjust the size of thread block

2025-03-12 14:33:36 +08:00

CMakeLists.txt

[CK TILE] Toy example - basic add

2025-03-11 16:23:12 +08:00

remod.py

introducing ck_tile! (#1216)

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