grimoire / TorchMPSCustomOpsDemo

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

About

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

License:Apache License 2.0


Languages

Language:C++ 37.0%Language:Objective-C++ 31.2%Language:Python 30.0%Language:Metal 1.8%