ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-03 05:01:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,464 Commits 734 Branches 37 Tags
14c3cfb1c6c67a34074fc3ee802ecb29c0e20d85
Commit Graph

2 Commits

Author SHA1 Message Date
rocking
0023f01ab0 [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>
 2024-10-07 14:25:53 +08:00
rocking
cb13839425 layernorm2d forward (#1339) 
* Add layernorm2d forward

* Refind file path

* clang format

* Exclude ck_tile op from all

* use add_executable instead

* refactor layernorm2d_fwd example

---------

Co-authored-by: carlushuang <carlus.huang@amd.com>
 2024-06-24 08:45:52 +08:00