metal-kernel

Umiejętność do implementacji natywnych kerneli Metal dla PyTorch na urządzeniach Apple Silicon. Prowadzi Cię przez dodawanie obsługi MPS do nowych operatorów oraz migrację istniejących implementacji z MPSGraph na natywny Metal. Obejmuje konfigurację dyspatchera w native_functions.yaml, implementację kerneli Metal w infrastrukturze c10/metal/ oraz stub'y po stronie hosta. Idealna dla deweloperów rozszerzających PyTorch o operacje GPU na Macu.

1. Zainstaluj umiejętność w swoim środowisku Claude lub Copilot, wskazując repozytorium pytorch/pytorch. 2. Otwórz plik aten/src/ATen/native/native_functions.yaml i zlokalizuj wpis operatora, którą chcesz zaimplementować lub zmigrować. 3. Dodaj lub zaktualizuj wpis dyspatchera, wskazując MPS jako cel — użyj formatu strukturalnego (structured: True) dla nowych operatorów lub skonsoliduj istniejące wpisy MPSGraph. 4. Utwórz plik kernela Metal w aten/src/ATen/native/mps/kernels/ — umiejętność pomoże Ci napisać kod wykorzystujący natywne możliwości Metal z infrastruktury c10/metal/, unikając MPSGraph. 5. Zaimplementuj stub po stronie hosta w aten/src/ATen/native/mps/operations/, który będzie łączyć logikę hosta z kernelem Metal. 6. Przetestuj implementację na urządzeniu Apple Silicon, weryfikując poprawność wyników i wydajność w porównaniu z wersjami CPU lub CUDA.