pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

https://pytorch.org/TensorRT

BSD 3-Clause "New" or "Revised" License

2.41k stars 333 forks source link

Open cehongwang opened 4 days ago

cehongwang commented 4 days ago

Description

Please review my changes for adding kwarg support for dynamo.compile