pytorch
5436134f - [MPS] Move the View ops to a separate file and reduce the number of graphs created (#80491)

Commit
3 years ago
[MPS] Move the View ops to a separate file and reduce the number of graphs created (#80491) This is dependent on the PR to go in first: https://github.com/pytorch/pytorch/pull/79939 Remove the data_ptr from the View Graph key which reduces the number of graphs created significantly. Don't wait when copying from MPS to MPS tensors Pull Request resolved: https://github.com/pytorch/pytorch/pull/80491 Approved by: https://github.com/malfet
Author
Committer
Parents
Loading