mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
By adding `mps` to the list of accelerators that can work with CPU scalars Fixes `GPUTests.test_masked_fill_promotion_mps` Pull Request resolved: https://github.com/pytorch/pytorch/pull/152268 Approved by: https://github.com/kulinseth, https://github.com/dcci, https://github.com/Skylion007 ghstack dependencies: #152266 |
||
|---|---|---|
| .. | ||
| linalg | ||
| nn | ||
| special | ||
| __init__.py | ||
| _conversions.py | ||
| fft.py | ||