search

qqaatw / pytorch-mps-ops-coverage

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

Add Torchvision functions in the matrix #2

Open ligaz opened 11 months ago

ligaz commented 11 months ago

Hey @qqaatw,

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

qqaatw commented 11 months 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!