Set model parameters to requires_grad=False (#3272)

jcaip · web-flow · commit 2a894913a9ae · 2025-11-03T12:30:09.000-05:00
* Set model parameters to requires_grad=False Disable gradient computation for model parameters to fix test failing after pytorch/pytorch#166367 * Fix formatting of requires_grad assignment
diff --git a/test/sparsity/test_marlin.py b/test/sparsity/test_marlin.py
@@ -39,6 +39,8 @@ def setUp(self):
             .half()
             .cuda()
         )
+        for param in self.model.parameters():
+            param.requires_grad = False
 
     @pytest.mark.skipif(not torch.cuda.is_available(), reason="Need CUDA available")
     @skip_if_rocm("ROCm enablement in progress")

Original file line number	Diff line number	Diff line change
`@@ -39,6 +39,8 @@ def setUp(self):`
`39`	`39`	`.half()`
`40`	`40`	`.cuda()`
`41`	`41`	`)`
	`42`	`+ for param in self.model.parameters():`
	`43`	`+ param.requires_grad = False`
`42`	`44`
`43`	`45`	`@pytest.mark.skipif(not torch.cuda.is_available(), reason="Need CUDA available")`
`44`	`46`	`@skip_if_rocm("ROCm enablement in progress")`