search

qqaatw / pytorch-mps-ops-coverage

PyTorch MPS backend Operators Coverage
Apache License 2.0
8 stars 0 forks source link

Add Torchvision functions in the matrix #2

Open ligaz opened 1 year ago

ligaz commented 1 year ago

Hey @qqaatw,

🙏 Thanks for providing the MPS operators coverage matrix. Is it possible to add all functions from Torchvision as well?

qqaatw commented 1 year ago

Hi, Torchvision ops are outside PyTorch and therefore not included in the matrix automatically. If you would like to have them in the matrix, you can manually register them in config.yml and submit a PR here.

Thank you!