TensorRT icon indicating copy to clipboard operation
TensorRT copied to clipboard
verified publisher icon pytorch

Allow Model Export Test Parallelism

Open leimao opened this issue 1 month ago • 2 comments

Description

Fixes #3934

Type of change

  • Bug fix (non-breaking change which fixes an issue)

Checklist:

  • [x] My code follows the style guidelines of this project (You can use the linters)
  • [x] I have performed a self-review of my own code
  • [x] I have commented my code, particularly in hard-to-understand areas and hacks
  • [ ] I have made corresponding changes to the documentation
  • [ ] I have added tests to verify my fix or my feature
  • [x] New and existing unit tests pass locally with my changes
  • [ ] I have added the relevant labels to my PR in so that relevant reviewers are notified

leimao avatar Nov 27 '25 04:11 leimao

@lanluo-nvidia please review

narendasan avatar Dec 01 '25 20:12 narendasan

@narendasan @lanluo-nvidia I tested the parallelism in my local environment on A100 and it worked fine.

leimao avatar Dec 02 '25 02:12 leimao