feat: add return_lse support for SDPA and SAGEV1 kernel — enable sequence-parallel (Ring Attention) merge #1991
a32543254
approved these changes
on 2026-07-03
Potential fix for pull request finding
2a710c5d
Potential fix for pull request finding
3f109b58
Potential fix for pull request finding
a0105260
chensuyue
added this to the 0.14.0 milestone 4 days ago
Add support for LSE output in SDPA and SAGEV1 kernels
1386f702
[pre-commit.ci] auto fixes from pre-commit.com hooks
e6c4a525
fix: adjust tolerance for SAGEV1 quantization noise in LSE tests
d3b4a3ee
Potential fix for pull request finding
483b5ee8
feat: enhance SAGE API with additional arguments for improved flexibi…
410deac9
refactor: improve code readability by formatting function signatures …
0c9da6e4
fix
6fc1a9ca
[pre-commit.ci] auto fixes from pre-commit.com hooks
53355118
chensuyue
merged
b481cebb
into main 4 days ago
chensuyue
deleted the sdpa_ring branch 4 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub