Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Huge performance gap between TVM and TRT on Stable Diffusion v1.5 #46

Open
felixslu opened this issue Jul 18, 2023 · 1 comment
Open

Comments

@felixslu
Copy link

felixslu commented Jul 18, 2023

GPU: Nvidia RTX 3090TI.

  1. Firstly, I use the log db in the repo, it gives me 3.7s to get the result.
  2. Then, I tried to tuning myself using meta-schedule(with trial count set to 50,000), it gives me 2.5s.

But, on TensorRT v8.6, for one iteration of unet, it gives me only 25ms, rather than 96ms with TVM(USE_CUBLAS =ON ; USE_CUDNN =ON; CUDA Version 12.1)

I wonder why the latency gap of stable diffusion model is so huge between TVM and TensorRT.
BTW, a few weeks ago, I got a different result between TVM and TRT, where my in-house model auto-tuned by TVM performs a wonderful infer latency (almost nearby TensorRT8.5).

Do you have any ideas about it? Thanks advance.

@felixslu felixslu changed the title benchmarks of stable diffusion v1.5 between TVM and TRT Huge performance gap between TVM and TRT on Stable Diffusion v1.5 Jul 18, 2023
@Civitasv
Copy link

Same for me.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants