ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-02 04:31:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
feb9a2bd9b50da9d449e5931e936d527a0db89fe
composable_kernel/include/ck_tile/ops/layernorm2d/kernel
History
valarLip 8ef8a994e7 [CK_TILE] add more stride for layernorm to support un-continuous Tensor (#1650) 
* [CK_TILE] add more stride for layernorm to support un-continuous Tensor

* align CK coding style

* extend strides to layernrom expample

* clang-format...
2024-11-11 16:02:28 +08:00
..
layernorm2d_fwd_kernel.hpp
[CK_TILE] add more stride for layernorm to support un-continuous Tensor (#1650)
2024-11-11 16:02:28 +08:00