[BugFix] test_mla_fp8.py fails on Cublas 12.9 (#11360)

Signed-off-by: Liu-congo <1502632128@qq.com>
This commit is contained in:
Liu-congo
2025-10-11 12:14:24 +08:00
committed by GitHub
parent eae9a9fb9d
commit c80a96dae9
2 changed files with 31 additions and 8 deletions

View File

@@ -263,6 +263,17 @@ def is_flashinfer_available():
return importlib.util.find_spec("flashinfer") is not None and is_cuda()
def is_nvidia_cublas_cu12_version_ge_12_9():
"""
temporary fix for issue #11272
"""
try:
installed_version = version("nvidia-cublas-cu12")
except PackageNotFoundError:
return False
return pkg_version.parse(installed_version) >= pkg_version.parse("12.9")
def random_uuid() -> str:
return str(uuid.uuid4().hex)