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

chore: fix use_cache flag #2965

Closed peri044 closed 1 day ago

peri044 commented 4 days ago

Description

Fix use_cache flag usage. We could actually remove use_cache flag and use cache if a user provides cache_file. But this would affect the existing workflows and backwards compatibility. So, changes are intentionally minimal here.

Fixes https://github.com/pytorch/TensorRT/issues/2960

Type of change

Please delete options that are not relevant and/or add your own.

Checklist:

narendasan commented 4 days ago

Better not change APIs in the legacy system, the fix seems fine