Skip to content

Commit b1ced92

Browse files
committed
skip if no cuda
1 parent c121731 commit b1ced92

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

test/integration/test_integration.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1515,6 +1515,7 @@ def forward(self, x):
15151515
model(x_in)
15161516

15171517
@parameterized.expand(list(itertools.product(["cuda"], COMMON_DTYPES)))
1518+
@unittest.skipIf(not torch.cuda.is_available(), "Need CUDA available")
15181519
def test_autoquant_min_sqnr(self, device, dtype):
15191520
m, k, n = 128, 128, 128
15201521
example_input = torch.randn(m, k, device=device, dtype=dtype)

0 commit comments

Comments
 (0)