From 3a0640ae93ba5b710d5fdfb7d6b60cb56f7e944e Mon Sep 17 00:00:00 2001
From: Anurag Dixit <a.dixit91@gmail.com>
Date: Wed, 20 Apr 2022 23:49:54 -0700
Subject: [PATCH] fix: Adding truncate_long_and_double to ptq tests

Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>
---
 tests/py/test_ptq_dataloader_calibrator.py | 1 +
 1 file changed, 1 insertion(+)

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,