update: simple_sparse_parameter

Author: kozistr

tests/utils.py

@@ -54,9 +54,10 @@ def simple_sparse_parameter(require_grad: bool = True) -> Tuple[torch.Tensor, to
     weight = torch.randn(5, 1).requires_grad_(require_grad)
     weight_sparse = weight.detach().requires_grad_(require_grad)
 
-    weight.grad = torch.rand_like(weight)
-    weight.grad[0] = 0.0
-    weight_sparse.grad = weight.grad.to_sparse()
+    if require_grad:
+        weight.grad = torch.rand_like(weight)
+        weight.grad[0] = 0.0
+        weight_sparse.grad = weight.grad.to_sparse()
 
     return weight, weight_sparse