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
36c7ce4e0eef86df186f8d796d7e177b8b13df92
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