[Perf] Optimize RMSNorm backward with staged reduction and vec8 I/O#855

coderfeli merged 5 commits into

ROCm:mainROCm/FlyDSL:mainfrom

jhinpan:perf/rmsnorm-bwd-two-stagejhinpan/FlyDSL-lab:perf/rmsnorm-bwd-two-stageCopy head branch name to clipboard