composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 13:11:25 +00:00

Files

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

* Adding seed and offset pointer support to the philox random number generator.

* Separating seed and offset pointer checks with different condition statements.

* Changes include, adding support for device seed and offset pointers, union is used to store seed/offset values and device pointers to minimize device SGPRs.

* Correcting a typo in the readme file

* Re-format files using remod.py

* Use STL type for API parameters

* Use simpler struct design for drop_seed & drop_offset

* Undo unnecessary changes

* Sync kargs style for fmha_fwd.hpp/.cpp

* Use templated union to reduce code

* Use structured binding to make code more readable

---------

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

2024-10-05 02:48:47 +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

layernorm2d forward (#1339 )

2024-06-24 08:45:52 +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 )