pytorch
8d1ff9fc - [MPS] Remove checks that incur unnecessary syncs on GPU with tensor.item() (#82505)

Commit
2 years ago
[MPS] Remove checks that incur unnecessary syncs on GPU with tensor.item() (#82505) ### Description <!-- What did you change and why was it needed? --> ### Issue <!-- Link to Issue ticket or RFP --> ### Testing <!-- How did you test your change? --> Pull Request resolved: https://github.com/pytorch/pytorch/pull/82505 Approved by: https://github.com/razarmehr, https://github.com/albanD
Author
Committer
Parents
  • aten/src/ATen/native
    • File
      Onehot.cpp
    • mps/operations
      • File
        Distributions.mm