composable_kernel

mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00

Files

dummycoderfe 7ba8518112 [Ck tile] layernorm2d fwd optimize (#1637 )

* optimze small N case using vec io and using rcp div

* [Ck_tile] layernorm, add param to control fastdiv; change generate codes and test pass

* [Ck_tile] fix blockSize compute in Generic2dBlockShape

* [Ck_tile]fix kfastfdiv template style

* [Ck_tile] layernorm, fix stype in review

---------

Co-authored-by: dummycoderfe <noplydummmycoder@163.com>

[ROCm/composable_kernel commit: 686a58a912]

2024-11-08 12:28:23 +08:00

generic_2d_block_shape.hpp

[Ck tile] layernorm2d fwd optimize (#1637 )

2024-11-08 12:28:23 +08:00

README.md

introducing ck_tile! (#1216 )

2024-04-15 19:27:12 -05:00

tensor_layout.hpp

introducing ck_tile! (#1216 )

2024-04-15 19:27:12 -05:00

README.md

common

this folder is designed not to be included directly by use, e.g. if use include ck_tile/ops/fmha.hpp, then everything under common should also be included.

to achieve this we will duplicate the header include path under common to other module under ops/* inside remod.py. for internal developer, you can also include ck_tile/ops/common.hpp for convenience. (and so does external users...)