Hugging Face's logo

kernels-community
/
rmsnorm

Kernels
Model card Files
xet
 Community
1

Add Metal (Apple Silicon) build variants

#1
by robtaylor-chipflow - opened
base: refs/heads/main
←
from: refs/pr/1
Discussion Files changed
+142
-0
robtaylor-chipflow

Add pre-built Metal kernel variants for Apple Silicon Macs.

Build variants:

  • torch210-metal-aarch64-darwin
  • torch29-metal-aarch64-darwin

These enable GPU-accelerated fused RMS normalization on MPS (Metal Performance Shaders) backend, tested on M1/M2/M3/M4 with macOS 14+.

74/74 tests passing across all dtypes and configurations.

Source: https://github.com/robtaylor/fused-rms-norm
Add Metal (Apple Silicon) build variantsaf102ff6
medmekk

Hey @robtaylor-chipflow , could you please open a pr here : https://github.com/huggingface/kernels-community instead ?

Cannot merge
This branch has merge conflicts in the following files:
  • .gitattributes

Sign up or log in to comment