This commit is contained in:
Alan Li 2026-08-16 01:40:26 +08:00 committed by GitHub
commit 38e1e37170
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions

View File

@ -1983,6 +1983,14 @@ def supports_nvfp4_compute(device=None):
if props.major < 10:
return False
# cuBLAS FP4 matmul kernels require CUDA 13+, see #11864
try:
cuda_version_major = int(torch.version.cuda.split(".")[0])
except (AttributeError, ValueError):
return False
if cuda_version_major < 13:
return False
return True
def supports_mxfp8_compute(device=None):