PyTorch 2.14에서 MPS의 index_add, index_select, conv3d, argmin/argmax 등이 Apple MPSGraph 경로를 떠나 손으로 짠 Metal 커널로 옮겨갑니다. 연산마다 붙던 컴파일 비용을 빼고, 스레드·메모리 접근을 직접 조율해 커널 기동 지연을 줄이는 쪽이 핵심입니다.

• index_add·index_select·conv3d·argmin/argmax가 네이티브 Metal로 이동합니다.

• MPSGraph 컴파일 비용을 덜어 기동 지연을 줄입니다.

• Apple Silicon에서 해당 연산이 자주 쓰이는 학습·추론 경로에 바로 닿습니다.

원문 요약은 DAKER에서 확인할 수 있습니다.

https://daker.ai/community/mps-index-add-conv3d-native-metal-kernel-latency