fix(packaging): AUT-1323 align dev package and TE revision - #6296
Closed
svcnemo-autobot wants to merge 4 commits into
Closed
fix(packaging): AUT-1323 align dev package and TE revision#6296svcnemo-autobot wants to merge 4 commits into
svcnemo-autobot wants to merge 4 commits into
Conversation
Add megatron.post_training and megatron.post_training.* to pyproject.toml's [tool.setuptools.packages.find] include list. Move modelopt imports to lazy (inside functions) so the module is importable without modelopt installed. Same fix as PR NVIDIA#6290 for the dev branch. Signed-off-by: svcnemo-autobot <svcnemo-autobot@nvidia.com>
Contributor
|
/ok to test d1a503b |
Pin fast-hadamard-transform to f134af63 and FlashMLA to b7643bd in the DSV4 GB200 recipe YAML. Unpinned clones of mutable branch HEADs can execute arbitrary future remote code on CI runners. Signed-off-by: svcnemo-autobot <svcnemo-autobot@nvidia.com>
Signed-off-by: svcnemo-autobot <svcnemo-autobot@nvidia.com>
Signed-off-by: svcnemo-autobot <svcnemo-autobot@nvidia.com>
svcnemo-autobot
changed the title
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
3 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Background
Bridge dev validation requires one consistent Transformer Engine revision across MCore and Bridge.
What changed
megatron.post_trainingpackaging and lazy-import fixes.cfda3834and admit its2.18.xversion in the dev constraint.uv.lock.Details
Bridge PR #5326 will pin this exact head after its approval mirror exists.
Tested
uv lock;uv lock --check— 325 packages>=2.18.0,<2.19.0admits2.18.0+cfda3834f031cf87remains inpyproject.tomloruv.lockgit diff --check