jakpiase e7b6286441 Add interwave scheduler for gemm mem pipeline (#1647) ...

* add interwave scheduler for gemm mem pipeline

* Fix merge artifacts.

* Refactor unit tests.

* Switch to interwave scheduler for mem example

---------

Co-authored-by: Adam Osewski <19374865+aosewski@users.noreply.github.com>
Co-authored-by: Adam Osewski <Adam.Osewski@amd.com>

2024-11-27 18:25:07 +01:00

..

01_fmha

[CK_TILE] Fix incorrect computation of group mode PagedAttention (#1688)

2024-11-26 20:37:54 +08:00

02_layernorm2d

[CK_TILE] add more stride for layernorm to support un-continuous Tensor (#1650)

2024-11-11 16:02:28 +08:00

03_gemm

Add interwave scheduler for gemm mem pipeline (#1647)

2024-11-27 18:25:07 +01: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] support rmsnorm and related fusion (#1605)

2024-10-30 15:22:56 +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] smoothquant (#1617)

2024-11-01 13:51:56 +08:00

11_add_rmsnorm2d_rdquant

[Ck_tile] smoothquant (#1617)

2024-11-01 13:51:56 +08:00

12_smoothquant

support max3 in smoothquant and add+ rmsnorm + rdquant (#1654)

2024-11-27 05:01:15 +08:00

13_moe_sorting

[CK_TILE] fused-moe first version (#1634)

2024-11-26 11:14:56 +08:00

14_moe_smoothquant

[CK_TILE]Moe update index (#1672)

2024-11-25 13:12:35 +08:00

15_fused_moe

[CK_TILE] fused-moe first version (#1634)

2024-11-26 11:14:56 +08:00

CMakeLists.txt

[CK_TILE] fused-moe first version (#1634)

2024-11-26 11:14:56 +08:00

remod.py

introducing ck_tile! (#1216)

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