fix: keep PPO token stats consistent under context parallelism - #990

Merged

garrett4wade merged 3 commits into

areal-project:mainfrom

yash27-lab:codex/fix-cp-token-stats

Mar 6, 2026

Commits on Mar 6, 2026

fix: keep PPO token stats consistent under context parallelism
yash27-lab
committed
5ba8c32
test: cover infer_token_denominator branches
yash27-lab
committed
6ea0a13
fix: avoid padded tree input_ids for n_tokens
yash27-lab
committed
4c4d3b8