ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
bf33df4e6e4b0fe1bb821faedb7512cd9e976c64
composable_kernel/include/ck_tile/ops
History
rocking bf33df4e6e 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>

[ROCm/composable_kernel commit: cb13839425]
2024-06-24 08:45:52 +08:00
..
common
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
epilogue
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
fmha
Fix FA bwd alibi+causal NaN errors (#1352)
2024-06-20 09:50:53 -05:00
gemm
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00
layernorm2d
layernorm2d forward (#1339)
2024-06-24 08:45:52 +08:00
reduce/block
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
welford
layernorm2d forward (#1339)
2024-06-24 08:45:52 +08:00
common.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
epilogue.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
fmha.hpp
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00
gemm.hpp
CK Tile FA Training kernels (#1286)
2024-06-04 13:12:45 -05:00
layernorm2d.hpp
layernorm2d forward (#1339)
2024-06-24 08:45:52 +08:00
reduce.hpp
introducing ck_tile! (#1216)
2024-04-15 19:27:12 -05:00
welford.hpp
layernorm2d forward (#1339)
2024-06-24 08:45:52 +08:00