ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 10:09:41 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
283920969ff4031676f52d0f0898a7bfa4231d87
composable_kernel/include/ck/tensor_operation/gpu
History
rocking 283920969f Backward of gamma and beta for layernorm and groupnorm (#1013) 
* Add layernorm backward reference code

* Add groupnorm backward reference code

* Add example

* clang format

* Fixc bug of reference layernorm and groupnorm

* Fix naming

* Refine naming

* Add device op for normalization bwd gamma and beta

* Refine template parameter

* Add bwd gamma & beta of kernel

* 1. Add groupnorm example
2. Refine layernorm naming

* Narrow down the static check for performance

* Refine variable name

[ROCm/composable_kernel commit: 1db7560365]
2023-11-10 18:02:03 +08:00
..
block
Add fp8 @ bf8 gemm support and example (#933)
2023-10-02 16:39:03 -05:00
device
Backward of gamma and beta for layernorm and groupnorm (#1013)
2023-11-10 18:02:03 +08:00
element
Add Gemm instances for performance improvement (#1018)
2023-11-07 09:09:58 -06:00
grid
Backward of gamma and beta for layernorm and groupnorm (#1013)
2023-11-10 18:02:03 +08:00
thread
add vector_type support into thread_copy_v3r1 (#969)
2023-10-13 15:11:43 -05:00
warp
Clean DTYPES conditions in CMake (#974)
2023-10-18 11:14:14 -05:00