Skip to content

Commit 846b8cf

Browse files
committed
fix: Adding truncate_long_and_double to ptq tests
Signed-off-by: Anurag Dixit <[email protected]>
1 parent cf5af74 commit 846b8cf

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tests/py/test_ptq_dataloader_calibrator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ def test_compile_script(self):
6565
"inputs": [torchtrt.Input([1, 3, 32, 32])],
6666
"enabled_precisions": {torch.float, torch.int8},
6767
"calibrator": self.calibrator,
68+
"truncate_long_and_double": True,
6869
"device": {
6970
"device_type": torchtrt.DeviceType.GPU,
7071
"gpu_id": 0,

0 commit comments

Comments
 (0)