Skip to content

Reshape instead of view in TiledFusedLogitsLoss - #8362

Merged
stas00 merged 2 commits into
deepspeedai:masterfrom
alanhuangyoo:fix/tiled-logits-loss-reshape
Sep 6, 2026
Merged

Reshape instead of view in TiledFusedLogitsLoss#8362
stas00 merged 2 commits into
deepspeedai:masterfrom
alanhuangyoo:fix/tiled-logits-loss-reshape

Assert the gradients in the tiled-logits layout test

ac714ff
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Sep 4, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO