feat(attention): add architecture-owned SM12x FA4 kernels (#32991)

Browse Source

...

This commit is contained in:

Gabriel Wu

2026-08-12 23:58:42 -07:00

committed by GitHub

parent ebc144ce3f

commit 889c2f31aa

16 changed files with 10449 additions and 198 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

test/registered/kernels/ops/attention/test_flash_attention_4_sm120.py

+2160

File diff suppressed because it is too large Load Diff