fix: Added truncate_long_and_double

Signed-off-by: Anurag Dixit <[email protected]>

Author: andi4191

tests/py/test_ptq_to_backend.py

@@ -40,6 +40,7 @@ def setUp(self):
                         "inputs": [torchtrt.Input([1, 3, 32, 32])],
                         "enabled_precisions": {torch.float, torch.half, torch.int8},
                         "calibrator": self.calibrator,
+                        "truncate_long_and_double": True,
                         "device": {
                             "device_type": torchtrt.DeviceType.GPU,
                             "gpu_id": 0,

tests/py/test_ptq_trt_calibrator.py

@@ -106,6 +106,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,

tests/py/test_trt_intercompatibility.py

@@ -16,6 +16,7 @@ def setUp(self):
     def test_pt_to_trt(self):
         compile_spec = {
             "inputs": [torchtrt.Input(self.input.shape)],
+            "truncate_long_and_double": True,
             "device": {
                 "device_type": torchtrt.DeviceType.GPU,
                 "gpu_id": 0,