pytorch/torch/_refs
Nikita Shulga bb680b5a87 [MPSInductor] Fix masked_fill decomp (#152268)
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
2025-04-27 15:50:46 +00:00
..
linalg Remove guard_size_oblivious from vector_norm decomposition. (#148809) 2025-04-10 16:19:00 +00:00
nn PEP585: More UP006 fixes (#146392) 2025-02-20 06:18:13 +00:00
special [Inductor][CPP] fix torch logit decomposition (#145576) 2025-01-27 19:37:51 +00:00
__init__.py [MPSInductor] Fix masked_fill decomp (#152268) 2025-04-27 15:50:46 +00:00
_conversions.py [BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765) 2024-07-31 10:42:50 +00:00
fft.py PEP585 update - torch/_C torch/_decomp torch/_lazy torch/_library torch/_numpy torch/_prims torch/_refs torch/_strobelight (#145102) 2025-01-18 20:47:12 +00:00