We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 43fab56 commit 3a0640aCopy full SHA for 3a0640a
tests/py/test_ptq_dataloader_calibrator.py
@@ -65,6 +65,7 @@ def test_compile_script(self):
65
"inputs": [torchtrt.Input([1, 3, 32, 32])],
66
"enabled_precisions": {torch.float, torch.int8},
67
"calibrator": self.calibrator,
68
+ "truncate_long_and_double": True,
69
"device": {
70
"device_type": torchtrt.DeviceType.GPU,
71
"gpu_id": 0,
0 commit comments