Skip to content
Open
Show file tree
Hide file tree
Changes from 14 commits
Commits
Show all changes
20 commits
Select commit Hold shift + click to select a range
11c3ed2
Add cuDNN score_mod attention path
vcherepanov-nv May 8, 2026
eb35191
Avoid BHSD copies in score_mod attention
vcherepanov-nv May 11, 2026
57ce106
Test relative position score_mod attention
vcherepanov-nv May 11, 2026
e6ba0ea
Test softcap score_mod attention
vcherepanov-nv May 11, 2026
dcb6b49
Run score_mod graphs on current CUDA stream
vcherepanov-nv May 11, 2026
fefcbe7
Add PyTorch score_mod execution plan cache
vcherepanov-nv May 12, 2026
ac4c60d
Fix score_mod cache edge cases
vcherepanov-nv May 12, 2026
6446825
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 13, 2026
58a5fb5
Fix score_mod callback graph cache keys
vcherepanov-nv May 15, 2026
c00a0b7
Address score_mod review feedback
vcherepanov-nv May 15, 2026
a8ed67e
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 15, 2026
e2a69e1
Fix score_mod lambda cache keys
vcherepanov-nv May 15, 2026
96f8ab2
Address flex attention review feedback
vcherepanov-nv May 19, 2026
e11cc23
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 19, 2026
f0f4f7b
Address flex attention backend review feedback
vcherepanov-nv May 21, 2026
9420479
Validate score_mod bprop tensor inputs
vcherepanov-nv May 21, 2026
db4e7c6
Add flex-attn tests to QA scripts
vcherepanov-nv May 22, 2026
6ca49cf
Add lint directive for score_mod tensor type checks
vcherepanov-nv May 22, 2026
a93df53
Alias dataclass field import in attention utils
vcherepanov-nv May 22, 2026
45ae2db
Skip softcap flex attention tests before sm90
vcherepanov-nv May 23, 2026
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Loading
Loading