Cherry pick Bump to pytorch 25.05 container along with TE update (13899) into r2.4.0#14145
Merged
chtruong814 merged 2 commits intor2.4.0from Jul 7, 2025
Merged
Cherry pick Bump to pytorch 25.05 container along with TE update (13899) into r2.4.0#14145chtruong814 merged 2 commits intor2.4.0from
Bump to pytorch 25.05 container along with TE update (13899) into r2.4.0#14145chtruong814 merged 2 commits intor2.4.0from
Conversation
* Update base container to be pytorch:25.05-py3 Signed-off-by: Charlie Truong <[email protected]> * Update TE to 2.4 Signed-off-by: Charlie Truong <[email protected]> * Remove torch accelerator patch Signed-off-by: Charlie Truong <[email protected]> * Update triton patch Signed-off-by: Charlie Truong <[email protected]> * Bump TE and Mcore commits Signed-off-by: Charlie Truong <[email protected]> * Fix triton patch Signed-off-by: Charlie Truong <[email protected]> * Fix triton patch Signed-off-by: Charlie Truong <[email protected]> * No fail fast Signed-off-by: Charlie Truong <[email protected]> * Update trt-llm to 0.20.0 Signed-off-by: Charlie Truong <[email protected]> * Fix test_sched_config_parse_reduce_on_plateau Signed-off-by: Charlie Truong <[email protected]> * Add no build isolation to TE Signed-off-by: Charlie Truong <[email protected]> * Update trt-llm dependencies Signed-off-by: Charlie Truong <[email protected]> * Update manifest Signed-off-by: Charlie Truong <[email protected]> * Revert "Enable LoRA for TELinear layers (#13929)" This reverts commit 7d9f40f. * update mcore with wd_mult key fix Signed-off-by: oliver könig <[email protected]> * Revert "Revert "Enable LoRA for TELinear layers (#13929)"" This reverts commit 5a1da6c. Signed-off-by: Charlie Truong <[email protected]> * Fix nemo install Signed-off-by: Charlie Truong <[email protected]> * Fix nemo install Signed-off-by: Charlie Truong <[email protected]> * Fix export image build Signed-off-by: Charlie Truong <[email protected]> * Remove unnecessary sed for torch_tensorrt Signed-off-by: Charlie Truong <[email protected]> * Update TE and Mcore commits Signed-off-by: Charlie Truong <[email protected]> * Add optional tests Signed-off-by: Charlie Truong <[email protected]> * Fix install Signed-off-by: Charlie Truong <[email protected]> * Ensure test script arg types are correct for top_p and top_k Signed-off-by: Charlie Truong <[email protected]> * Increase export deploy timeouts Signed-off-by: Charlie Truong <[email protected]> * Skip failing test_rnnt_logprobs_random after pytorch bump Signed-off-by: Charlie Truong <[email protected]> * Skip coverage artifact config-3.12.py Signed-off-by: Charlie Truong <[email protected]> * Include more config files ot exclude during coverage Signed-off-by: Charlie Truong <[email protected]> * Update dependencies Signed-off-by: Charlie Truong <[email protected]> * Ensure top_p is float in nemo_export test script Signed-off-by: Charlie Truong <[email protected]> * Set Optional_L2_Speech_Batch_Size_OOMptimizer_Canary to truly be optional Signed-off-by: Charlie Truong <[email protected]> * Fix top_k and top_p types in megatronllm_deployable Signed-off-by: Charlie Truong <[email protected]> * Revert "Skip failing test_rnnt_logprobs_random after pytorch bump" This reverts commit c6c3a76. Signed-off-by: Charlie Truong <[email protected]> * Fix optional export test Signed-off-by: Charlie Truong <[email protected]> * Revert unnecessary changes Signed-off-by: Charlie Truong <[email protected]> --------- Signed-off-by: Charlie Truong <[email protected]> Signed-off-by: oliver könig <[email protected]> Co-authored-by: Alexandros Koumparoulis <[email protected]> Co-authored-by: oliver könig <[email protected]>
ko3n1g
requested review from
chtruong814,
pablo-garay and
thomasdhc
as code owners
Signed-off-by: Charlie Truong <[email protected]>
chtruong814
approved these changes
Jul 7, 2025
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
2 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.
beep boop [🤖]: Hi @chtruong814 👋,