We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
test_galore_quant.py
1 parent 173d38f commit 87ff45aCopy full SHA for 87ff45a
test/quantization/test_galore_quant.py
@@ -15,6 +15,14 @@
15
except ImportError: # noqa: F401
16
pytest.skip("triton is not installed", allow_module_level=True) # noqa: F401
17
import torch
18
+
19
+# Skip entire test if CUDA is not available or ROCM is enabled
20
+if not torch.cuda.is_available() or torch.version.hip is not None:
21
+ pytest.skip(
22
+ "CUDA is not available/ ROCM support is under development",
23
+ allow_module_level=True,
24
+ )
25
26
from bitsandbytes.functional import (
27
create_dynamic_map,
28
dequantize_blockwise,
0 commit comments