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

danielhanchen merged 2 commits into

unslothai:mainunslothai/unsloth:mainfrom

nole69:fix/cross-entropy-device-mismatch-contdnole69/unsloth:fix/cross-entropy-device-mismatch-contdCopy head branch name to clipboard