ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-02 04:37:02 +00:00
Code Issues Packages Projects Releases Wiki Activity
2,680 Commits 777 Branches 40 Tags
c4199307ecd257a8e1a0da53e6dcd7597cc53568
Commit Graph

6 Commits

Author SHA1 Message Date
Mohsen Saffari
c4199307ec Change from NCHW to MHWC based on old-ck and manage verifying for c > 1 2025-12-02 15:47:46 +00:00
Mohsen Saffari
3194b653f7 add tiling, pipeline, follow layernorm2d 2025-11-28 18:29:10 +00:00
Mohsen Saffari
b732595d9f add kSaveMeanInvStd, kUpdateMovingAverage in Traits 2025-11-27 15:06:13 +00:00
Mohsen Saffari
5c8e8684ec add gamma, bias to the simple kernel 2025-11-27 13:32:18 +00:00
Mohsen Saffari
c7afd540d9 correct the batch normalization algorithm to calc reduce over NxHxW 2025-11-25 16:46:00 +00:00
Mohsen Saffari
5fa3ec2b37 Some refactoring for simple batchnorm kernel and example 2025-11-25 13:14:58 +00:00