Skip to content

Add example for custom MPS kernels on macOS #124

@0x00b1

Description

@0x00b1

Would be great to have a simple example showing how to build and use custom Metal (MPS) kernels with torch.utils.cpp_extension on Apple Silicon. Right now there’s nothing covering MPS, and it’s not obvious how to compile .metal files or register them with PyTorch. Even a minimal example (like an elementwise add) and a short note in the README would help a lot.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions