Repository navigation
Commit bb0fe53
authored
fix(deps): align pyproject DeepEP pin with the container build (NVIDIA-NeMo#3608)
DeepEP is pinned in two places that nothing keeps in sync:
docker/Dockerfile ARG DEEPEP_COMMIT = 42144303 (DeepEP NVIDIA-NeMo#638, 2026-05-20)
pyproject.toml deep_ep.rev = 7febc6e2 (DeepEP NVIDIA-NeMo#580, 2026-02-09)
The Dockerfile pin was bumped to pick up DeepEP NVIDIA-NeMo#638 (pads
max_num_of_tokens_per_rank to the combine-kernel chunk size, fixing the
HybridEP JIT static_assert), but the uv source was left behind. Since the
`moe` extra is not part of `all`, the container installs DeepEP only from
the Dockerfile source build -- so CI never exercised the stale pin, while
anyone running `uv sync --extra moe` or `pip install nemo_automodel[moe]`
outside the container got a DeepEP three months older than the one every
recipe was validated against.
Bump the uv source to 42144303 and update the declared
`[[tool.uv.dependency-metadata]]` version to `v1.2.1+4214430` to match
(deep_ep builds as `1.2.1+$(git rev-parse --short HEAD)`; uv trusts this
string instead of building during resolution). Both lock files are
refreshed accordingly.
Add tests/unit_tests/test_deepep_pin_consistency.py so the next bump
cannot land half-applied: it fails when the two pins disagree, and when
the declared version stops naming the pinned commit.
Signed-off-by: Alexandros Koumparoulis <akoumparouli@nvidia.com>1 parent 1e40293 commit bb0fe53
4 files changed
Lines changed: 126 additions & 10 deletions
File tree
- docker/common
- tests/unit_tests
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
116 | 116 | | |
117 | 117 | | |
118 | 118 | | |
119 | | - | |
| 119 | + | |
120 | 120 | | |
121 | 121 | | |
122 | 122 | | |
| |||
1661 | 1661 | | |
1662 | 1662 | | |
1663 | 1663 | | |
1664 | | - | |
1665 | | - | |
| 1664 | + | |
| 1665 | + | |
1666 | 1666 | | |
1667 | 1667 | | |
1668 | 1668 | | |
| |||
4265 | 4265 | | |
4266 | 4266 | | |
4267 | 4267 | | |
4268 | | - | |
| 4268 | + | |
4269 | 4269 | | |
4270 | 4270 | | |
4271 | 4271 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
301 | 301 | | |
302 | 302 | | |
303 | 303 | | |
304 | | - | |
| 304 | + | |
305 | 305 | | |
306 | 306 | | |
307 | 307 | | |
| |||
360 | 360 | | |
361 | 361 | | |
362 | 362 | | |
363 | | - | |
| 363 | + | |
364 | 364 | | |
365 | 365 | | |
366 | 366 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
| 10 | + | |
| 11 | + | |
| 12 | + | |
| 13 | + | |
| 14 | + | |
| 15 | + | |
| 16 | + | |
| 17 | + | |
| 18 | + | |
| 19 | + | |
| 20 | + | |
| 21 | + | |
| 22 | + | |
| 23 | + | |
| 24 | + | |
| 25 | + | |
| 26 | + | |
| 27 | + | |
| 28 | + | |
| 29 | + | |
| 30 | + | |
| 31 | + | |
| 32 | + | |
| 33 | + | |
| 34 | + | |
| 35 | + | |
| 36 | + | |
| 37 | + | |
| 38 | + | |
| 39 | + | |
| 40 | + | |
| 41 | + | |
| 42 | + | |
| 43 | + | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
| 47 | + | |
| 48 | + | |
| 49 | + | |
| 50 | + | |
| 51 | + | |
| 52 | + | |
| 53 | + | |
| 54 | + | |
| 55 | + | |
| 56 | + | |
| 57 | + | |
| 58 | + | |
| 59 | + | |
| 60 | + | |
| 61 | + | |
| 62 | + | |
| 63 | + | |
| 64 | + | |
| 65 | + | |
| 66 | + | |
| 67 | + | |
| 68 | + | |
| 69 | + | |
| 70 | + | |
| 71 | + | |
| 72 | + | |
| 73 | + | |
| 74 | + | |
| 75 | + | |
| 76 | + | |
| 77 | + | |
| 78 | + | |
| 79 | + | |
| 80 | + | |
| 81 | + | |
| 82 | + | |
| 83 | + | |
| 84 | + | |
| 85 | + | |
| 86 | + | |
| 87 | + | |
| 88 | + | |
| 89 | + | |
| 90 | + | |
| 91 | + | |
| 92 | + | |
| 93 | + | |
| 94 | + | |
| 95 | + | |
| 96 | + | |
| 97 | + | |
| 98 | + | |
| 99 | + | |
| 100 | + | |
| 101 | + | |
| 102 | + | |
| 103 | + | |
| 104 | + | |
| 105 | + | |
| 106 | + | |
| 107 | + | |
| 108 | + | |
| 109 | + | |
| 110 | + | |
| 111 | + | |
| 112 | + | |
| 113 | + | |
| 114 | + | |
| 115 | + | |
| 116 | + | |
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.
0 commit comments