[PyTorch][Flash Attn] Add fallback import for FA3 #2806

Merged

vcherepanov-nv merged 3 commits intoNVIDIA:mainfrom

eattia-nvidia:fix_fa3_import_fallback

Apr 8, 2026

Commits on Mar 25, 2026

[PyTorch][Flash Attn] Add fallback import for FA3 when flash_attn_interface.py is outside flash_attn_3 package
eattia-nvidia
committed

Commits on Mar 26, 2026

Merge branch 'main' into fix_fa3_import_fallback
eattia-nvidia
authored
[PyTorch][Flash Attn] Use find_spec for FA3 import and add diagnostic warning
eattia-nvidia
committed