trunk/7ed5a45a0670bc1c39b3f5b88f343ed69d12a22a: [MPS] Migrate hardtanh_backward to Metal (#200791)
- PyTorch 近 90 天出现 2301 次
- PyTorch 近 90 天第 2271 次发版
- 上一次：同一天稍早 · trunk/f73d515f10de16ccbbff74a71b14409385f5046a
发生了什么
Fixes #200790 Replaces the MPSGraph implementation of hardtanh_backward with a native Metal kernel. Performance Layout Shape Dtype MPSGraph (µs) Metal (µs) Speedup contiguous 1024 float32 12.73 5.16 2.47x contiguous 1024 float16 12.62 5.03 2.51x contiguous 1024 bfloat16 13.14 5.05 2.60x contiguous 1024x1024 float32 54.46 55.35 0.98x contiguous 1024x1024 float16 28.26 28.73 0.98x contiguous 1024x1024 bfloat16 28.25 2…
摘要按规则整理自下方来源原文