pytorch
42d6543c - [bc-breaking] Dispatch index_put with boolean mask argument to masked_fill (#61612)

Commit
3 years ago
[bc-breaking] Dispatch index_put with boolean mask argument to masked_fill (#61612) Summary: https://github.com/pytorch/pytorch/issues/57515 Based on ngimel 's branch, with a few tweaks to determine when to copy value tensors to device memory/additional tests. bc-breaking note: Previously, if in `x[index]=value` `value` was a 0-d tensor with device different from `x`'s device, it resulted in a RuntimeError. Now this case is handled by copying `value` to the correct device. Pull Request resolved: https://github.com/pytorch/pytorch/pull/61612 Reviewed By: mrshenli Differential Revision: D29753491 Pulled By: ngimel fbshipit-source-id: 3fba14f4c2b9b136b50af020f9c1eda88f7373b0
Author
eqy eqy
Parents
Loading