feat(bucketing_and_reordering): support bucketing and reordering for FSDP training - #495

Closed

lipengfei1409 wants to merge 29 commits into

chenyifan/joint_passesfrom

lipengfei/feat/bucheting_and_reordering

Commits on Dec 21, 2025

feat(dispatch): use aot_autograd as the simpler dispatcher, and add joint stage
yfchen-byted
committed
7381557
fix(inference): force fw_compiler=inference_compiler under inference mode
yfchen-byted
committed
41748bc
fix(test_cases): temp skip failed test cases under new dispatcher
yfchen-byted
committed
d94c628

Commits on Dec 22, 2025

ci(new_dispatch): add ci for new dispatch
yfchen-byted
committed
82426c8
Merge branch 'master' into chenyifan/new_dispatch
yfchen-byted
authored
886f0cd

Commits on Jan 4, 2026

Merge branch 'master' into chenyifan/new_dispatch
yfchen-byted
authored
9224db3

Commits on Jan 28, 2026

Merge branch 'master' into chenyifan/new_dispatch
yfchen-byted
authored
8516a6d

Commits on Feb 6, 2026

feat: add test_trainer for test training with xpu_graph
lipengfei1409
committed
eaf145b

Commits on Feb 9, 2026

feat: support bucketing and reordering in fsdp
lipengfei1409
committed
e571ceb
fix: fix some bug
lipengfei1409
committed
4b0085d
chore: delete some docstring
lipengfei1409
committed
293c012
chore: delete not used file
lipengfei1409
committed
5277cc3
chore: fix pre-commit errors
lipengfei1409
committed
63340bd
chore: fix some bugs
lipengfei1409
committed
dbc2c7c
chore: fix pre-commit errors
lipengfei1409
committed
0bd617c
chore: rename item in config and set False as default
lipengfei1409
committed
e17f8ac
chore: delete args
lipengfei1409
committed
cde3918
chore: add some comment
lipengfei1409
committed
8a58228
style(lintcheck): run lint check
yfchen-byted
committed
adf732d
chore: merge chenyifan/new_dispatch branch
lipengfei1409
committed
4e2c766
feat: support reshard_after_forward
lipengfei1409
committed
281d0cd
chore: merge the latest branch
lipengfei1409
committed
21b52a2

Commits on Feb 25, 2026

test: add test for reshard_after_forward
lipengfei1409
committed
e5f27f9
chore: apply ruff
lipengfei1409
committed
377a6df
chore: modify tests
lipengfei1409
committed
cca3f31
chore: modify something
lipengfei1409
committed
2b59e16

Commits on Feb 26, 2026

chore: add some test code
lipengfei1409
committed
a33ce48
chore: fix lint errors
lipengfei1409
committed
3906caf

Commits on Feb 27, 2026

Merge branch 'chenyifan/joint_passes' into lipengfei/feat/bucheting_and_reordering
lipengfei1409
authored
df5bfe3