fix: Adding truncate_long_and_double to ptq tests

Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>
pytorch · May 2, 2022 · 3a0640a · 3a0640a
1 parent 43fab56
commit 3a0640a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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,