Skip to content

Add THD-format per-rank batch fetch with padding mask for sequence packing - #6625

Merged
ilml merged 2 commits into
NVIDIA:mainfrom
ilml:split/3386-04-thd-batch-fetch
Aug 19, 2026
Merged

Add THD-format per-rank batch fetch with padding mask for sequence packing#6625
ilml merged 2 commits into
NVIDIA:mainfrom
ilml:split/3386-04-thd-batch-fetch

Address review: drop the TE wrapper and reuse broadcast_scalars

e8cad32
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Aug 18, 2026 in 1s

DCO

All commits are signed off!