Skip to content

Commit 9a9ecde

Browse files
authored
Skip galore test if not cuda (#2003)
Summary: fixing CI before branch cut Test Plan: python test/quantization/test_galore_quant.py and CI Reviewers: Subscribers: Tasks: Tags:
1 parent e4eff3a commit 9a9ecde

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

test/quantization/test_galore_quant.py

+2
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@
3838

3939

4040
@pytest.mark.skip("skipping for now, see comments below")
41+
@pytest.mark.skipif(not torch.cuda.is_available(), reason="Need CUDA available")
4142
@pytest.mark.parametrize(
4243
"dim1,dim2,dtype,signed,blocksize",
4344
TEST_CONFIGS,
@@ -89,6 +90,7 @@ def test_galore_quantize_blockwise(dim1, dim2, dtype, signed, blocksize):
8990
TEST_CONFIGS,
9091
)
9192
@skip_if_rocm("ROCm enablement in progress")
93+
@pytest.mark.skipif(not torch.cuda.is_available(), reason="Need CUDA available")
9294
def test_galore_dequant_blockwise(dim1, dim2, dtype, signed, blocksize):
9395
g = torch.randn(dim1, dim2, device="cuda", dtype=dtype) * 0.01
9496

0 commit comments

Comments
 (0)