Unverified Commit b4003d11 authored by Michael Goin's avatar Michael Goin Committed by GitHub
Browse files

Check if bitblas is installed during support check (#17572)


Signed-off-by: default avatarmgoin <mgoin64@gmail.com>
parent 292fc59d
...@@ -71,6 +71,15 @@ def _check_bitblas_supported( ...@@ -71,6 +71,15 @@ def _check_bitblas_supported(
f"Only group_sizes = {BITBLAS_SUPPORTED_GROUP_SIZES} " f"Only group_sizes = {BITBLAS_SUPPORTED_GROUP_SIZES} "
"are supported.") "are supported.")
# Finally, check if bitblas is installed
try:
import bitblas
if bitblas.__version__ < MINIMUM_BITBLAS_VERSION:
raise ImportError("bitblas version is wrong. Please "
f"install bitblas>={MINIMUM_BITBLAS_VERSION}")
except ImportError:
return False, "BitBLAS is not installed."
return True, None return True, None
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment