flash-attention-explorer / src /attention_utils.py

Commit History

fix: Add fallback SDPA benchmark when attention layer fails
685194e

a0y0346 commited on

feat: Use real HuggingFace model attention layers for benchmarks
374d38b

a0y0346 commited on