trunk/7ed5a45a0670bc1c39b3f5b88f343ed69d12a22a: [MPS] Migrate hardtanh_backward to Metal (#200791)
- PyTorch: 2298 events in the last 90 days
- PyTorch: 2268th Release in the last 90 days
- Previous: earlier the same day · trunk/f73d515f10de16ccbbff74a71b14409385f5046a
What happened
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…
Summary assembled by rule from the sources below