feat(pt_expt): multi-task training support#5397

Merged

wanghan-iapcm merged 31 commits intodeepmodeling:masterfrom

wanghan-iapcm:feat-pt-expt-multi-task

Apr 20, 2026

Commits on Apr 15, 2026

feat(pt_expt): multi-task training support
Han Wang
committed

Commits on Apr 16, 2026

fix(dpmodel): wrap fparam/aparam reshape with descriptive ValueError
Han Wang
committed
fix: address CodeQL findings in PR #5397
Han Wang
committed
fix(pt_expt): access unwrapped module in _compile_model for DDP compat
Han Wang
committed
test(pt_expt): add DDP + torch.compile training tests
Han Wang
committed
feat(pt_expt): use inductor+dynamic for training compile
Han Wang
committed
test(pt_expt): port silut activation + repformers accessors from #5393
Han Wang
committed
test(pt_expt): assert virial in compile correctness tests
Han Wang
committed
test(pt_expt): port silu compile and varying-natoms tests from #5393
Han Wang
committed
test(pt_expt): compare compiled vs uncompiled with varying natoms
Han Wang
committed
test(pt_expt): cover DPA2/DPA3 in varying-natoms compile correctness
Han Wang
committed
test(pt_expt): exercise DPA2 three-body branch in compile correctness
Han Wang
committed

Commits on Apr 17, 2026

fix(dpmodel): restore nf in reshapes to fix zero-atom and add silu_backward decomp
Han Wang
committed
fix: address CodeQL findings in PR #5397
Han Wang
committed
fix(pt): wrap fparam/aparam reshape with descriptive ValueError
Han Wang
committed

Commits on Apr 18, 2026

feat(pt_expt): reject DPA1/se_atten_v2 with attention at compile time
Han Wang
committed
fix(pt_expt): remove false DPA1 attention compile guard
Han Wang
committed
refactor(dpmodel): remove unused get_numb_attn_layer API
Han Wang
committed
fix(test): use real path for PT water data, remove unused API
Han Wang
committed
fix(pt_expt): rebuild FX graph after detach node removal to avoid segfaults
Han Wang
committed
fix(pt_expt): tune inductor options for compile training
Han Wang
committed
fix(pt_expt): disable DDPOptimizer to prevent compiled graph splitting
Han Wang
committed
fix(test): add .cpu() before .numpy() for GPU-compatible activation tests
Han Wang
committed
fix(pt_expt): revert inductor options that cause numerical divergence
Han Wang
committed
fix(test): make DDP tests device-adaptive instead of hardcoding CPU
Han Wang
committed
fix(test): correct freeze test docstrings to match dpa3 guard
Han Wang
committed
fix(pt_expt): move optimize_ddp into _compile_model, resolve test symlinks
Han Wang
committed
fix(test): backup/restore fparam.npy in TestFparam instead of deleting
Han Wang
committed
fix(test): skip DDP tests when NCCL is selected with fewer than 2 GPUs
Han Wang
committed

Commits on Apr 20, 2026