NVIDIA / NVIDIA/TensorRT

turning on refit increase the inferring time of trt engine

Open
#2,936 6 comments 0 reactions 1 assignee View on GitHub

@BowenFu is already working on this.

Since May 7, 2023.

triaged
Dominant language
C++
Stars
13.4k
Forks
2.4k
Avg merge
5d 3h
Merged PRs (30d)
2

Description

Hi, tensorRT team! I used trt to increase the inferring time of stable diffusion about 3 times, really awesome. However, cause building a tensorrt engine is time-consuming, I'd like to use refit to replace the parameters of a tensorrt engine when I have a new stable diffusion torch model with the same structure. Then I implemented a test on refit to see how it influenced the performance of a tensorrt engine, and I found without refit, the trt engine ran 30 steps for 1.16s, while it ran 30 steps for 2.02s with refit. It seems that turning on refit will increase the inferencing time of trt engine, is that right?

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.