pytorch/torch/_decomp
Natalia Gimelshein 112ec24f09 Fix device behavior for masked_fill (#82737)
Fixes #81018, based on #81036.
It will create graph break for cpu 0d tensor value due to .item() call (we could maybe specialize on that instead of breaking?), but otherwise it would create graph break due to synchronizing `to` call, so there's no way around :-(, and for number `value` argument we already should be specializing.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/82737
Approved by: https://github.com/Chillee
2022-08-04 15:47:56 +00:00
..
__init__.py prevent python view impls from getting registered to the meta key (#82007) 2022-07-27 17:15:05 +00:00
decompositions.py Fix device behavior for masked_fill (#82737) 2022-08-04 15:47:56 +00:00