We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 251fddb commit 8c62eb0Copy full SHA for 8c62eb0
test/dtypes/test_nf4.py
@@ -192,7 +192,6 @@ def test_smoketest_linear(self):
192
out1 = torch.nn.functional.linear(inp, a)
193
out2 = torch.nn.functional.linear(inp, a_nf4)
194
195
- @unittest.skipIf(torch.__version__.split('+')[0] == '2.2.1', "Broken on stable.")
196
@unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
197
def test_smoketest_linear_compile(self):
198
for dtype in [torch.bfloat16, torch.float16]:
0 commit comments