pytorch
b7bf953b - [MPS] Fix bernoulli for int types (#100946)

Commit
1 year ago
[MPS] Fix bernoulli for int types (#100946) <!-- copilot:summary --> ### <samp>🤖 Generated by Copilot at 069fd23</samp> This pull request enhances the MPS implementation of random operations in `Distributions.mm` and adds more dtype tests for the bernoulli distribution in `test_mps.py`. This improves the performance, correctness, and usability of the MPS backend for PyTorch. Fixes https://github.com/pytorch/pytorch/issues/100717 Pull Request resolved: https://github.com/pytorch/pytorch/pull/100946 Approved by: https://github.com/kulinseth
Author
Committer
Parents
Loading