We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b6ca41c commit f799b57Copy full SHA for f799b57
tests/contrib/test_prompter.py
@@ -87,7 +87,7 @@ def test_gradcheck(self, device):
87
def test_module(self):
88
...
89
90
- @pytest.mark.skipif(torch_version() in {"2.0.0", "2.0.1"}, reason="Not working on 2.0")
+ @pytest.mark.skipif(torch_version() in {"2.0.0", "2.0.1", "2.1.2"}, reason="Not working on 2.0.x and 2.1.x")
91
def test_dynamo(self, device, torch_optimizer):
92
dtype = torch.float32
93
batch_size = 1
0 commit comments