viable/strict/1792428192: [MPS] Migrate std_mean to metal (#201310)

发生了什么

Migrates std_mean to metal by reusing existing reduction primitives. Perf on m5 pro (no regressions): Shape Reduction dims Dtype Layout Strides std_mean speedup var_mean speedup (32,) all fp32 contiguous (1,) 3.36x 3.46x (32,) all bf16 contiguous (1,) 3.42x 3.46x (32,) all fp32 stride 2 (offset 1) (2,) 3.97x 4.01x (32,) all bf16 stride 2 (offset 1) (2,) 3.97x 3.99x (1024,) all fp32 contiguous (1,) 2.67x 2.60x (1024,…

摘要按规则整理自下方来源原文

来源