diff --git a/tests/py/test_ptq_dataloader_calibrator.py b/tests/py/test_ptq_dataloader_calibrator.py index d9f28cfd32..158a5425e8 100644 --- a/tests/py/test_ptq_dataloader_calibrator.py +++ b/tests/py/test_ptq_dataloader_calibrator.py @@ -65,6 +65,7 @@ def test_compile_script(self): "inputs": [torchtrt.Input([1, 3, 32, 32])], "enabled_precisions": {torch.float, torch.int8}, "calibrator": self.calibrator, + "truncate_long_and_double": True, "device": { "device_type": torchtrt.DeviceType.GPU, "gpu_id": 0,