pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
https://pytorch.org/TensorRT
BSD 3-Clause "New" or "Revised" License
2.6k stars 351 forks source link

📖 [Story] Optimize the launch overhead of TRT engine and pytorch kernels #3274

Open keehyuna opened 3 weeks ago

keehyuna commented 3 weeks ago

TL;DR

Runtime optimization in torch-tensorrt is crucial for maximizing model performance in real-world applications. This story tracks the effort to improve runtime performance.

Goal(s)

Tasks

### Tasks
- [ ] https://github.com/pytorch/TensorRT/pull/3276
- [ ] https://github.com/pytorch/TensorRT/issues/3277

Additional context

### Tasks