Skip to content

[BUG FIX] Optimize top-k mask construction: prevent unsafe gradient flow and eliminate unnecessary memory allocations #184

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
LoserCheems wants to merge 1 commit into
base: main
Choose a base branch
from

Fixes top‑k attention masking and safe casting

d7d87ef
Select commit
Loading
Failed to load commit list.
Sign in for the full log view
Open

[BUG FIX] Optimize top-k mask construction: prevent unsafe gradient flow and eliminate unnecessary memory allocations #184

Fixes top‑k attention masking and safe casting
d7d87ef
Select commit
Loading
Failed to load commit list.
add-reviewers
succeeded Oct 4, 2025 in 7s