[rmsnorm] Support FP32 weights with FP16/BF16 activations#884

coderfeli merged 1 commit into

ROCm:mainROCm/FlyDSL:mainfrom

jhinpan:feat/rmsnorm-mixed-weight-dtypejhinpan/FlyDSL-lab:feat/rmsnorm-mixed-weight-dtypeCopy head branch name to clipboard