search

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

🐛 [Bug] DataLoaderCalibrator cache check #2960

Closed lebionick closed 1 day ago

lebionick commented 6 days ago

Invalid check, so if cache_file is not None and use_cache is True it will raise an Exception. https://github.com/pytorch/TensorRT/blob/main/py/torch_tensorrt/ts/ptq.py#L112

peri044 commented 4 days ago

Thanks for catching this. https://github.com/pytorch/TensorRT/pull/2965 should resolve it.