Skip to content

Commit 3a0640a

Browse files
committed
fix: Adding truncate_long_and_double to ptq tests
Signed-off-by: Anurag Dixit <[email protected]>
1 parent 43fab56 commit 3a0640a

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

Diff for: tests/py/test_ptq_dataloader_calibrator.py

+1
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)