deepspeedai / deepspeedai/DeepSpeed

[BUG] DeepCompile: Training hang on random-sized inputs

Open
#7,611 24 comments 0 reactions 1 assignee View on GitHub

@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:

  1. The compute graph may not be exactly the same across ranks (e.g. some have specific padding while the others don't).
  2. Guard failure (due to tensor size change) on different ranks may occur at different iterations.

To Reproduce

  1. Download https://gist.github.com/eternalNight/3c2cf8c703f1e9e7742d3b7f9e1edae3
  2. Execute deepspeed --num_gpus=N openvla-like.py -c -r.

Expected behavior

DeepCompile works for variable-length training data.

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.