Merge 8f4e58d29f into a9ab2b62da
This commit is contained in:
commit
38e1e37170
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Reference in New Issue