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 853ded0 commit 566f3b8Copy full SHA for 566f3b8
vllm/model_executor/layers/quantization/utils/bitblas_utils.py
@@ -71,6 +71,15 @@ def _check_bitblas_supported(
71
f"Only group_sizes = {BITBLAS_SUPPORTED_GROUP_SIZES} "
72
"are supported.")
73
74
+ # Finally, check if bitblas is installed
75
+ try:
76
+ import bitblas
77
+ if bitblas.__version__ < MINIMUM_BITBLAS_VERSION:
78
+ raise ImportError("bitblas version is wrong. Please "
79
+ f"install bitblas>={MINIMUM_BITBLAS_VERSION}")
80
+ except ImportError:
81
+ return False, "BitBLAS is not installed."
82
+
83
return True, None
84
85
0 commit comments