deepspeedai / deepspeedai/DeepSpeed
[BUG] DeepCompile: Training hang on random-sized inputs
Open
@Shreyas-S-809 is already working on this.
Since Mar 17, 2026.
bug
training
- Dominant language
- Python
- Stars
- 43.1k
- Forks
- 5k
- Avg merge
- 4d 15h
- Merged PRs (30d)
- 112
Description
Description
Real-life training data may not be of the same size for every rank and at every iteration. When DeepCompile is active, training with variable-length data can hang because DeepCompile requires communication among the ranks during profiling, but:
- The compute graph may not be exactly the same across ranks (e.g. some have specific padding while the others don't).
- Guard failure (due to tensor size change) on different ranks may occur at different iterations.
To Reproduce
- Download https://gist.github.com/eternalNight/3c2cf8c703f1e9e7742d3b7f9e1edae3
- Execute
deepspeed --num_gpus=N openvla-like.py -c -r.
Expected behavior
DeepCompile works for variable-length training data.
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Assessment
This issue has not been assessed yet.