1f85f39e0a
This PR fixes the "Arch conditional MMA instruction used without targeting appropriate compute capability. Aborting." errors that occur when using FBGEMM on Blackwell GPUs (B200/B100, SM100). Changes: - Add stderr filters in import_fixes.py for CUTLASS/FBGEMM MMA errors - Add warning filters for various deprecation messages - Update check_fbgemm_gpu_version() to disable FBGEMM instead of raising an error when old versions are detected - Update test_has_fbgemm() in fp8.py to catch broader CUTLASS/CUDA errors and gracefully fall back to Triton kernels - Update loader_utils.py to disable FBGEMM instead of raising ValueError for old fbgemm_gpu versions The key behavior change is that FBGEMM errors no longer crash the script. Instead, FBGEMM is disabled and Triton kernels are used automatically. This allows Unsloth to work on SM100 GPUs where CUTLASS SM90 kernels fail, and also gracefully handles old FBGEMM versions. |
||
|---|---|---|
| .. | ||
| moe | ||
| __init__.py | ||
| cross_entropy_loss.py | ||
| fast_lora.py | ||
| flex_attention.py | ||
| fp8.py | ||
| geglu.py | ||
| layernorm.py | ||
| rms_layernorm.py | ||
| rope_embedding.py | ||
| swiglu.py | ||
| utils.py | ||