ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 02:27:57 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
35e652e2865fdcfbce2d7752b352ddf33eb4e838
composable_kernel/include/ck_tile/ops/layernorm2d/kernel
History
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
..
layernorm2d_fwd_kernel.hpp
[Ck tile] Support layernorm one pass (#1512)
2024-10-07 14:25:53 +08:00