Lightning Talk: Enhancements Made to MPS Backend in PyTorch for Applications Running...
summary
MPS backend qualifies as beta in PyTorch 2.0 with support for 300+ operators and expanded network coverage; profiling is integrated via OS signpost and Metal System Trace (MPSProfiler & Instruments) so developers can see blit calls, MPS execution and CPU fallbacks. Custom kernels compile as C++/Objective‑C extensions using APIs like get_command_buffer, get_dispatch_q, synchronize and commit_and_continue to eliminate CPU fallbacks (softshrink replaced) and reduce copies. MacOS 14 results show up to 5x speedup across TorchBench workloads.
AI-written summary · about
chapters (4)
- 0:00MPS backend overview
- 2:01Profiler and monitoring
- 4:35Custom kernel development
- 7:27APIs and performance results
related talks
08mSponsored Session: Lightning Talk: PyTorch in Production: Boosting LLM Training and...
17mLightning Thunder: Supercharged PyTorch for Modern Hardware
24mAdvancements in High-Performance AI/ML through PyTorch's Python Compiler
29mKeynote
47mGet to Know PyTorch - A Cutting Edge Open Source AI Framework for Deep Learning
26mThunder: Distribute and Optimize Your PyTorch Models With Zero...
55mAPAC/ US Opening Talks
31mPyTorch 2.1 - New Developments
