Skip to content

ci: Fix TensorRT engine build error for vision models#8479

Merged
mc-nv merged 2 commits into
r25.10from
spolisetty/tri-212-dlfw2510triton-l0_model_control_stress-failed-failed-to-open
Oct 29, 2025
Merged

ci: Fix TensorRT engine build error for vision models#8479
mc-nv merged 2 commits into
r25.10from
spolisetty/tri-212-dlfw2510triton-l0_model_control_stress-failed-failed-to-open

Conversation

@pskiran1

@pskiran1 pskiran1 commented Oct 29, 2025

Copy link
Copy Markdown
Member

What does the PR do?

This PR provides a workaround to successfully build TensorRT engines for vision models in CI.
We need a follow-up ticket to address the following warning.

DeprecationWarning: You are using the legacy TorchScript-based ONNX export. Starting in PyTorch 2.9, the new torch. export-based ONNX exporter will be the default. To switch now, set dynamo=True in torch.onnx.export. This new exporter supports features like exporting LLMs with DynamicCache. We encourage you to try it and share feedback to help improve the experience. Learn more about the new export logic: https://pytorch.org/docs/stable/onnx_dynamo.html. For exporting control flow: https://pytorch.org/tutorials/beginner/onnx/export_control_flow_model_to_onnx_tutorial.html.
  torch.onnx.export(

Checklist

  • PR title reflects the change and is of format <commit_type>: <Title>
  • Changes are described in the pull request.
  • Related issues are referenced.
  • Populated github labels field
  • Added test plan and verified test passes.
  • Verified that the PR passes existing CI.
  • Verified copyright is correct on all changed files.
  • Added succinct git squash message before merging ref.
  • All template sections are filled out.
  • Optional: Additional screenshots for behavior/output changes with before/after.

Commit Type:

Check the conventional commit type
box here and add the label to the github PR.

  • build
  • ci
  • docs
  • feat
  • fix
  • perf
  • refactor
  • revert
  • style
  • test

Related PRs:

Where should the reviewer start?

Test plan:

  • CI Pipeline ID: 37522279

Caveats:

Background

Related Issues: (use one of the action keywords Closes / Fixes / Resolves / Relates to)

  • closes GitHub issue: #xxx

@pskiran1 pskiran1 added the PR: ci Changes to our CI configuration files and scripts label Oct 29, 2025
@pskiran1
pskiran1 marked this pull request as ready for review October 29, 2025 15:46
@pskiran1
pskiran1 requested a review from mc-nv October 29, 2025 16:33
@mc-nv
mc-nv merged commit 33b6a14 into r25.10 Oct 29, 2025
2 of 3 checks passed
@mc-nv
mc-nv deleted the spolisetty/tri-212-dlfw2510triton-l0_model_control_stress-failed-failed-to-open branch October 29, 2025 17:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

PR: ci Changes to our CI configuration files and scripts

Development

Successfully merging this pull request may close these issues.

3 participants