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

Files

Po Yen Chen e7ff49610b [CK_TILE] Update example README files & fix script compatibility issue (#1548 )

* Fix text alignment of ArgParser::print()

* Update example README files

* Clarify make-ck-dev.sh <arch> usage

* Only keep some of the argument from '-?' output

* Undo command line output changes in README

* Only keep existing argument on doc and update description

* Fix text alignment

* Make cmake-ck-*.sh compatible with 'sh' command

[ROCm/composable_kernel commit: 0c094daa7e]

2024-10-08 10:45:12 +08:00

CMakeLists.txt

layernorm2d forward (#1339 )

2024-06-24 08:45:52 +08:00

layernorm2d_fwd.cpp

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

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

layernorm2d_fwd.hpp

layernorm2d forward (#1339 )

2024-06-24 08:45:52 +08:00

README.md

[CK_TILE] Update example README files & fix script compatibility issue (#1548 )

2024-10-08 10:45:12 +08:00

README.md

Layernorm2D forward

This folder contains example for Layernorm2D forward using ck_tile tile-programming implementation.

build

# in the root of ck_tile
mkdir build && cd build
# you can replace <arch> with the appropriate architecture (for example gfx90a or gfx942) or leave it blank
sh ../script/cmake-ck-dev.sh  ../ <arch>
make tile_example_layernorm2d_fwd -j

This will result in an executable build/bin/tile_example_layernorm2d_fwd

example

args:
          -m    m dimension (default:3328)
          -n    m dimension (default:4096)
          -e    epsilon (default:1e-5)
          -v    cpu validation or not (default:1)
       -prec    precision (default:fp16)