viable/strict/1790279487: [MPS] migrate rounding ops (#198497)

发生了什么

Migrates rounding ops such as floor , ceil , trunc . Perf: dtype shape floor (µs) ceil (µs) trunc (µs) fp32 32x32 16.4 → 2.3 18.2 → 2.3 15.2 → 2.4 fp32 256x256 15.5 → 2.8 16.2 → 2.8 18.3 → 2.8 fp32 1024x1024 32.1 → 18.2 28.4 → 18.2 28.7 → 18.2 fp32 4096x4096 541.6 → 503.5 546.6 → 504.3 546.4 → 528.2 fp32 4096x4096 .t() 527.3 → 502.7 521.6 → 497.6 518.0 → 502.3 fp32 4096x4096 [:, :4000] 520.2 → 486.9 513.4 → 485.5 52…

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

来源