amend

pytorch · Sep 11, 2024 · 231908c · 231908c
1 parent 5eb1892
commit 231908c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_nn.py b/test/test_nn.py
@@ -3699,7 +3699,7 @@ def __setattr__(self, key, value):
 @pytest.mark.skipif(not torch.cuda.is_available(), reason="CUDA is not available")
 @pytest.mark.parametrize("compiled", [True, False])
 class TestCudaGraphs:
-    @pytest.fixture(scope="class")
+    @pytest.fixture(scope="class", autouse=True)
     def _set_cuda_device(self):
         device = torch.get_default_device()
         torch.set_default_device("cuda:0")