grimoire / TorchMPSCustomOpsDemo Public

Notifications You must be signed in to change notification settings
Fork 1
Star 9

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

Apache-2.0 license

9 stars 1 fork Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 13 Commits
csrc/pytorch		csrc/pytorch
.gitignore		.gitignore
LICENCE		LICENCE
README.md		README.md
custom_add_test.py		custom_add_test.py
setup.py		setup.py

Repository files navigation

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.

Apache-2.0 license

Report repository

Releases

No releases published

Packages

No packages published

Languages