-
metal-kernel：把 PyTorch 在 Apple Silicon 上的内核，从 MPSGraph 拉回原生 Metal
用 Apple Silicon 跑 PyTorch 的人，多半对 MPS 后端又爱又恨。爱的是它确实让 M1/M2/M3 那几颗芯片有了存在感，恨的是它总在关键时刻掉链子，要么算子不支持，要么数值对不上，要么性能比 CPU 还拉胯。 问题的根子，PyTorch 官方自己也承认，出在 MPSGraph 这套实现方式上。MPSGraph 是 Apple 的高层图 API，PyTorch 早期为了快速铺…- 1.1k
- 0