Skip to content

[FIX] Move loss and n_items to logits device in fast_cross_entropy_loss loss for multi-GPU support #4063

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

Merged
danielhanchen merged 2 commits into from
Feb 15, 2026

Apply suggestion from @gemini-code-assist[bot]

5f68761
Select commit
Loading
Failed to load commit list.
Merged

[FIX] Move loss and n_items to logits device in fast_cross_entropy_loss loss for multi-GPU support #4063

Apply suggestion from @gemini-code-assist[bot]
5f68761
Select commit
Loading
Failed to load commit list.

Workflow runs completed with no jobs