[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>
2026-04-20 06:49:15 +00:00 · 2024-10-07 14:25:53 +08:00
parent c24fae2346
commit 0023f01ab0
3 changed files with 263 additions and 96 deletions
--- a/example/ck_tile/02_layernorm2d/layernorm2d_fwd.cpp
+++ b/example/ck_tile/02_layernorm2d/layernorm2d_fwd.cpp
@@ -35,7 +35,9 @@ float layernorm2d_fwd(layernorm2d_fwd_traits t,
                                                                    YDataType,
                                                                    MeanDataType,
                                                                    InvStdDataType,
-                                                                    Shape>;
+                                                                    Shape,
+                                                                    true,
+                                                                    true>;

        using Kernel = ck_tile::Layernorm2dFwd<PipelineProblem>;