Update on "torch.sgn for complex tensors"

anjali411 · anjali411 · commit 0fed8c8e1e17 · 2020-09-21T20:26:32.000-07:00
resolves #36323 by adding `torch.sgn` for complex tensors. `torch.sgn` returns `x/abs(x)` for `x != 0` and returns `0 + 0j` for `x==0` also updates the backward definition of `torch.div`, `torch.abs` Differential Revision: [D23460526](https://our.internmc.facebook.com/intern/diff/D23460526) [ghstack-poisoned]
diff --git a/torch/testing/_internal/common_nn.py b/torch/testing/_internal/common_nn.py
@@ -3966,7 +3966,6 @@ def padding3d_circular(input, pad):
         target_fn=lambda: torch.randn(15, 10).gt(0).double(),
         reference_fn=lambda i, t, m: -(t * i.log() + (1 - t) * (1 - i).log()).sum() /
             (i.numel() if get_reduction(m) else 1),
-        check_gradgrad=False,
         check_bfloat16=TEST_WITH_ROCM,
     ),
     dict(
@@ -3978,7 +3977,6 @@ def padding3d_circular(input, pad):
         reference_fn=lambda i, t, m: -((t * i.log() + (1 - t) * (1 - i).log()) * get_weight(m)).sum() /
             (i.numel() if get_reduction(m) else 1),
         desc='weights',
-        check_gradgrad=False,
         check_bfloat16=TEST_WITH_ROCM,
     ),
     dict(
@@ -4328,7 +4326,6 @@ def padding3d_circular(input, pad):
         reference_fn=lambda i, t, m: -((t * i.log() + (1 - t) * (1 - i).log()) * get_weight(m)).sum() /
             (i.numel() if get_reduction(m) == 'mean' else 1),
         desc='scalar_weights',
-        check_gradgrad=False,
         check_bfloat16=TEST_WITH_ROCM,
     ),
     dict(
