💾 trunk/7120d05eddfb5563e592a89f83bcdee7baa4911c: [MPS] median and nanmedian to metal (#187060)
Nachrichtenbereich: 💾 Downloads
🔗 Quelle: github.com
Fixes #187017
median and nanmedian to metal. Perf:
Global reduction (torch.median(x) / torch.nanmedian(x))
config
dtype
before (us)
after (us)
speedup
global 1e6
f32
376.1
104.9
3.59x
global... [Weiterlesen]