grimoire / TorchMPSCustomOpsDemo

A demo about how to add custom MPS ops in PyTorch.

PyTorch MPS Custom Ops

This is a demo of mps custom ops for PyTorch.

installation:

python setup.py develop

test:

python custom_add_test.py