metal: cache x in registers in rms_single_row to avoid redundant global read#3754

zcbenz merged 2 commits into

ml-explore:mainml-explore/mlx:mainfrom

will-march:perf/rms-norm-cache-xwill-march/mlx:perf/rms-norm-cache-xCopy head branch name to clipboard