[Refactor] Align RMSNorm with current FlyDSL kernel style#902

coderfeli merged 4 commits into

ROCm:mainROCm/FlyDSL:mainfrom

jhinpan:perf/rmsnorm-fx-cleanupjhinpan/FlyDSL-lab:perf/rmsnorm-fx-cleanupCopy head branch name to clipboard