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

What is the precise meaning of cuDNN and TensorRT in the result table? #4

Closed
umiswing opened this issue Apr 23, 2023 · 1 comment
Closed

Comments

@umiswing
Copy link

Hello! I am interested in TVM's work on cutlass. I am a bit confused about the results table. Does the inclusion of cuDNN and TensorRT in the table mean that TVM is running models using the kernels/sub-graphs provided by cuDNN and TensorRT? If that is the case, why did TVM add support for cutlass when TensorRT is faster?

@masahi
Copy link
Owner

masahi commented Apr 23, 2023

Does the inclusion of cuDNN and TensorRT in the table mean that TVM is running models using the kernels/sub-graphs provided by cuDNN and TensorRT

Yes for cuDNN but no for TRT. The TRT result is for reference only, the results were obtained by running TRTorch on resnet or running the BERT demo in the TensorRT repo. And CUTLASS is generally faster than cuDNN.

@masahi masahi closed this as completed Apr 23, 2023
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