Qianfeng Zhang d7930cd541 Update IsFulleTileInsideMask() for kUseLocal is true situtation

2025-06-06 15:55:26 +00:00

..

01_fmha

add mask support in hdim=192/128 (#1999)

2025-03-21 18:28:43 +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

fix ck_tile/basic_gemm build error (#1988)

2025-03-20 22:01:14 -07:00

04_img2col

…

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

…

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

[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] return value with macro in ck_tile::kernel_launch API (#1982)

2025-03-20 11:00:29 -07:00

16_batched_gemm

[CK_TILE] Switch to universal gemm for batched and grouped gemms (#1919)

2025-03-20 11:17:04 +01:00

17_grouped_gemm

[CK_TILE] Switch to universal gemm for batched and grouped gemms (#1919)

2025-03-20 11:17:04 +01:00

18_hstu_attention

Update IsFulleTileInsideMask() for kUseLocal is true situtation

2025-06-06 15:55:26 +00:00

35_batched_transpose

add batched_transpose implement (#1660)

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

CMakeLists.txt

Initial reference implementation of hstu attention

2025-03-28 16:26:43 +00:00

remod.py

…