composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00

Files

Qianfeng e58d441138 Change in fwd-splitkv kernel to support num_splits=1 case (#1690 )

* Change in fwd-splitkv kernel to support num_splits=1 case

* Update in codegen fwd-splitkv to make num_splits > 1 cases pass

* Specify instance traits in dispatch

* Fix link error for fp8 kernels

---------

Co-authored-by: Po Yen Chen <PoYen.Chen@amd.com>

[ROCm/composable_kernel commit: ce2bdf42a9]

2024-11-25 12:31:38 +08:00

01_fmha

Change in fwd-splitkv kernel to support num_splits=1 case (#1690 )

2024-11-25 12:31:38 +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

[CK Tile] Improve the Layout, Padding, and Alignment features of CK Tile GEMM (#1651 )

2024-11-11 18:08:25 -08:00

04_img2col

[CK_TILE] Update example README files & fix script compatibility issue (#1548 )