Skip to content

fix: skip nonfinite fused AdamW updates - #152

Merged
yubofredwang merged 1 commit into
lightseekorg:mainfrom
Dogacel:fix/bf16-found-inf
Aug 4, 2026
Merged

fix: skip nonfinite fused AdamW updates#152
yubofredwang merged 1 commit into
lightseekorg:mainfrom
Dogacel:fix/bf16-found-inf

Conversation

@Dogacel

@Dogacel Dogacel commented Aug 4, 2026

Copy link
Copy Markdown
Collaborator

Sometimes a bad row / data point can corrupt the whole training. Previously it was gated by grad_norm > 0.0 check, however it forces a D2H sync, which slows down training. Current approach lets optimizer handle it internally.

image

@chatgpt-codex-connector

Copy link
Copy Markdown

Codex usage limits have been reached for code reviews. Please check with the admins of this repo to increase the limits by adding credits.
Repo admins can enable using credits for code reviews in their settings.

@yubofredwang
yubofredwang merged commit 89296a6 into lightseekorg:main Aug 4, 2026
0 of 2 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants