Emily Martins a3499e38b2 Add CK Tile Stream-K bf16 and fp16 examples ...

Addition of initial CK Tile Stream-K example for bf16 and fp16. These
examples are minimal. As more functionality and gtests are added for
Stream-K (coming in future PRs), these examples will be expanded.

2025-09-29 15:46:24 -06:00

..

01_fmha

[CK_TILE] Support f32 in FMHA (fwd and bwd) (#2836)

2025-09-27 18:03:48 +05:00

02_layernorm2d

[CK_TILE] Refine Generic2dBlockShape to fix ck_tile example 2,10,11,14 on rdna3 and 4 (#2795)

2025-09-10 08:29:20 +08:00

03_gemm

fix wp gemm bug when permuteN is false (#2935)

2025-09-26 13:28:54 -07:00

04_img2col

invoke script directly (#2687)

2025-08-19 00:23:07 -07:00

05_reduce

[CK_TILE] fix example reduces, permute and elementwise on gfx11 & gfx12 (#2810)

2025-09-11 12:41:20 +08:00

06_permute

[CK_TILE] fix example reduces, permute and elementwise on gfx11 & gfx12 (#2810)

2025-09-11 12:41:20 +08:00

09_topk_softmax

Fix latest AITER failure and add more AITER tests in CK CI. (#2782)

2025-09-04 13:44:00 -07:00

10_rmsnorm2d

[CK_TILE] Refine Generic2dBlockShape to fix ck_tile example 2,10,11,14 on rdna3 and 4 (#2795)

2025-09-10 08:29:20 +08:00

11_add_rmsnorm2d_rdquant

[CK_TILE] Refine Generic2dBlockShape to fix ck_tile example 2,10,11,14 on rdna3 and 4 (#2795)

2025-09-10 08:29:20 +08:00

12_smoothquant

[CK_TILE] Refine Generic2dBlockShape to fix ck_tile example 2,10,11,14 on rdna3 and 4 (#2795)

2025-09-10 08:29:20 +08:00

13_moe_sorting

Fix latest AITER failure and add more AITER tests in CK CI. (#2782)

2025-09-04 13:44:00 -07:00

14_moe_smoothquant

[CK_TILE] Refine Generic2dBlockShape to fix ck_tile example 2,10,11,14 on rdna3 and 4 (#2795)

2025-09-10 08:29:20 +08:00

15_fused_moe

fix(copyright header): add header to missing files (#2807)

2025-09-11 12:27:08 -07:00

16_batched_gemm

[CK_TILE] Fix example batched_gemm, grouped_gemm, gemm_multi_d, convolution on gfx11 & gfx12 (#2808)

2025-09-11 07:27:33 -07:00

17_grouped_gemm

Integrate Multi D GEMMs into Grouped GEMMs along with unit tests (#2923)

2025-09-26 09:59:58 -07:00

18_flatmm

[CK_TILE] Enable ck_tile tests on gfx11 and gfx12 (#2821)

2025-09-12 12:45:14 -07:00

19_gemm_multi_d

[CK_TILE] Fix example batched_gemm, grouped_gemm, gemm_multi_d, convolution on gfx11 & gfx12 (#2808)

2025-09-11 07:27:33 -07:00

20_grouped_convolution

[CK Tile] Implement Invoker pattern for remaining grouped convolution examples (#2894)

2025-09-24 10:22:38 +02:00

21_elementwise

Congma/ck tile/remove cpp 20 code (#2873)

2025-09-25 10:34:28 -07:00

22_gemm_multi_abd

[CK_TILE] Multiple-ABD GEMM example (#2788)

2025-09-19 01:14:11 +02:00

35_batched_transpose

Fix latest AITER failure and add more AITER tests in CK CI. (#2782)

2025-09-04 13:44:00 -07:00

38_block_scale_gemm

Weight Preshuffle Block Scale gemm support (#2877)

2025-09-29 12:46:37 -07:00

39_copy

fix(copyright header): add header to missing files (#2807)

2025-09-11 12:27:08 -07:00

40_streamk_gemm

Add CK Tile Stream-K bf16 and fp16 examples

2025-09-29 15:46:24 -06:00

CMakeLists.txt

Add CK Tile Stream-K bf16 and fp16 examples

2025-09-29 15:46:24 -06:00

remod.py

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00