Repository navigation

(fsdp2 dev.) - follow-up : Add training.static_shape_buckets: a few static lengths for compiled blocks - #929

Draft

yushengsu-thu wants to merge 9 commits into

sgl-project:mainfrom

yushengsu-thu:fsdp2/shape-buckets

Commits on Oct 2, 2026

Add optional FSDP2 training backend
yushengsu-thu
committed
3acf1ae
Fix FSDP2 parameter retention and root resharding
yushengsu-thu
committed
11841f7

Commits on Oct 4, 2026

Add training.compile_blocks for the FSDP2 backend
yushengsu-thu
committed
177dc64
Forward training.compile_blocks through the disaggregated launch path
yushengsu-thu
committed
b60fe73
Pass backend options to create_training_backend only when set
yushengsu-thu
committed
d8e3cf3
Add training.static_shapes: one batch length and anchor count per run
yushengsu-thu
committed
7815e57
Warn instead of rejecting compile_blocks without static_shapes
yushengsu-thu
committed
6311268
static_shapes: tolerate an anchor count above the candidate positions
yushengsu-thu
committed
7ce2eeb
Add training.static_shape_buckets: a few static lengths for compiled blocks
yushengsu-thu
committed
5cb8627