Skip to content

Commit 8c62eb0

Browse files
authored
Reenable nf4 compile smoke test (#101)
1 parent 251fddb commit 8c62eb0

File tree

1 file changed

+0
-1
lines changed

1 file changed

+0
-1
lines changed

test/dtypes/test_nf4.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,6 @@ def test_smoketest_linear(self):
192192
out1 = torch.nn.functional.linear(inp, a)
193193
out2 = torch.nn.functional.linear(inp, a_nf4)
194194

195-
@unittest.skipIf(torch.__version__.split('+')[0] == '2.2.1', "Broken on stable.")
196195
@unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
197196
def test_smoketest_linear_compile(self):
198197
for dtype in [torch.bfloat16, torch.float16]:

0 commit comments

Comments
 (0)