fix: Added truncate_long_and_double

Signed-off-by: Anurag Dixit <[email protected]>
pytorch · May 2, 2022 · 417c096 · 417c096
1 parent 3a0640a
commit 417c096
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/tests/py/test_ptq_to_backend.py b/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,

diff --git a/tests/py/test_ptq_trt_calibrator.py b/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,

diff --git a/tests/py/test_trt_intercompatibility.py b/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,