← Alle Claude SkillsClaude Skill · Development
Gids voor het schrijven van Metal Kernel
# PyTorch Metal/MPS Kernels Implementation Guide ## 1. Native Functions Registration ### native_functions.yaml Configuration ```yaml - name: my_op variants: function cpu: true cuda: true mps: true dispatch: CPU: my_op_cpu CUDA: my_op_cuda MPS: my_op_mps tags: pointwise - name: my_op.out variants: function cpu: true cuda: true mps: true dispatch: CPU: my_op_cpu_out CUDA: my_op_cuda_out MPS: my_op_mps_out tags: pointwise ``` ## 2. Host-Side Operator Implementation ### C++ Kernel Registration (Ops.
Bron: github.com/pytorch/pytorch/tree/main/.claude/skills/metal-kernel · Licentie: Other
Volledige uitleg, versies en download zijn beschikbaar voor leden — lidmaatschap is gratis.