pytorch
355a1c8c
- MPS: TopK raise an error if K>16 (#79677)
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
2 years ago
MPS: TopK raise an error if K>16 (#79677) * Error out in TopK when k>16. * Add a test case too. Fixes #78915 Pull Request resolved: https://github.com/pytorch/pytorch/pull/79677 Approved by: https://github.com/albanD
Author
kulinseth
Committer
pytorchmergebot
Parents
270c518b
Loading