pytorch
b654d149 - [MPS] Fix the argument error for tensor_split() test (#94234)

Commit
1 year ago
[MPS] Fix the argument error for tensor_split() test (#94234) The second tensor argument `tensor_indices_or_sections` of tensor_split() must be on CPU when testing it in TestConsistency. Otherwise it will error out. Pull Request resolved: https://github.com/pytorch/pytorch/pull/94234 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading