composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00

Files

rocking 9abf356d74 [Ck tile] Support layernorm one pass (#1512 )

* Fix compile error

* Add one pass pipeline

* Extract creating tile_window to operator()

* clang format

* reduce duplicated code

* do not hardcode

* Support padding in layernorm

---------

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

[ROCm/composable_kernel commit: 0023f01ab0]

2024-10-07 14:25:53 +08:00

01_fmha

Adding seed and offset pointer support to the philox random number generator. (#1523 )

2024-10-05 02:48:47 +08:00

02_layernorm2d

[Ck tile] Support layernorm one pass (#1512 )

2024-10-07 14:25:53 +08:00

03_gemm

Ck tile gemm padding dim (#1516 )

2024-09-18 11:32:29 -07:00

04_img2col

[CK_TILE] Image to Column kernel (#1532 )