/TorchMPSCustomOpsDemo

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

Primary LanguageC++Apache License 2.0Apache-2.0

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