pytorch
7a408c72 - [complex] `masked_fill`: Complex Autograd support and update masked_scatter skips. (#52035)

Commit
3 years ago
[complex] `masked_fill`: Complex Autograd support and update masked_scatter skips. (#52035) Summary: Now that `masked_fill` CUDA is migrated, skips on masked_scatter can be removed. Reference: https://github.com/pytorch/pytorch/issues/33152 **Note**: Have decreased the shape of Tensor for `masked_scatter` from (M, M) -> (S, S) and so on. With shapes of M : **96.53s** ``` test/test_ops.py ........................................ssssssssssss........................ssssssssssss........................ [100%] =============================================================== 88 passed, 24 skipped, 7981 deselected in 96.53s (0:01:36) ================================================================ ``` With shapes of S : **46.53s** ``` test/test_ops.py ........................................ssssssssssss........................ssssssssssss........................ [100%] ==================================================================== 88 passed, 24 skipped, 7981 deselected in 46.53s ===================================================================== ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/52035 Reviewed By: VitalyFedyunin Differential Revision: D26369476 Pulled By: anjali411 fbshipit-source-id: 7a79d5a609b0019f8fe9ce6452924dd33390dce1
Author
Parents
Loading