TorchMPSCustomOpsDemo

C++ ★ 9 updated 3y ago

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

No plain-English explanation yet — one is being written right now. Check back in a minute.