remove torch.no_grad from sample_inputs (#78076)
As per title
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78076
Approved by: https://github.com/Lezcano, https://github.com/ngimel
diff --git a/torch/testing/_internal/common_methods_invocations.py b/torch/testing/_internal/common_methods_invocations.py
index bd7628d..6686130 100644
--- a/torch/testing/_internal/common_methods_invocations.py
+++ b/torch/testing/_internal/common_methods_invocations.py
@@ -6656,23 +6656,22 @@
ks = (1, 3, 0)
for b, n, k, (left, upper, uni) in product(bs, ns, ks, product((True, False), repeat=3)):
- with torch.no_grad():
- if b == 1:
- A = make_arg((n, n)) if left else make_arg((k, k))
- B = make_arg((n, k))
- else:
- A = make_arg((b, n, n)) if left else make_arg((b, k, k))
- B = make_arg((b, n, k))
- if uni:
- # Not really necessary, but writing it for consistency
- A.diagonal(0, -2, -1).fill_(1.)
- else:
- d = A.diagonal(0, -2, -1)
- d[d.abs() < 1e-6] = 1.
- if upper:
- A.triu_()
- else:
- A.tril_()
+ if b == 1:
+ A = make_arg((n, n)) if left else make_arg((k, k))
+ B = make_arg((n, k))
+ else:
+ A = make_arg((b, n, n)) if left else make_arg((b, k, k))
+ B = make_arg((b, n, k))
+ if uni:
+ # Not really necessary, but writing it for consistency
+ A.diagonal(0, -2, -1).fill_(1.)
+ else:
+ d = A.diagonal(0, -2, -1)
+ d[d.abs() < 1e-6] = 1.
+ if upper:
+ A.triu_()
+ else:
+ A.tril_()
kwargs = {"upper": upper, "left": left, "unitriangular": uni}
if requires_grad:
for grad_A, grad_B in product((True, False), repeat=2):
@@ -9128,12 +9127,11 @@
t1 = _make_tensor(s, low=0)
t2 = _make_tensor(s, low=0)
- with torch.no_grad():
- if not li:
- i1.abs_()
- i2.abs_()
- t1.abs_()
- t2.abs_()
+ if not li:
+ i1.abs_()
+ i2.abs_()
+ t1.abs_()
+ t2.abs_()
yield (
i1, t1,
