William Lin ba75ad82db [refactor]: shared attention infra additions for QAT-compat (Attn-QAT 5/12) (#1383)

2026-05-23 16:19:24 -07:00

..

test_attention_metadata_base.py

[refactor]: shared attention infra additions for QAT-compat (Attn-QAT 5/12) (#1383)

2026-05-23 16:19:24 -07:00

test_flash_attn_cute_custom_op.py

[misc] FlashAttention 4 support (#1114)

2026-03-07 16:53:43 -08:00

test_flash_attn_default_custom_op.py

[perf]: register FA2/FA3 default flash_attn_func as a torch.library custom op (#1373)

2026-05-23 00:39:06 -07:00

test_flash_attn_no_pad_resolver.py

[refactor]: shared attention infra additions for QAT-compat (Attn-QAT 5/12) (#1383)

2026-05-23 16:19:24 -07:00