Skip to content

Commit 9ca3f62

Browse files
committed
Removing the int64 casting to int32 in harness.py since native int64 in supported in TRT10
1 parent 6afbb8d commit 9ca3f62

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

tests/py/dynamo/conversion/harness.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -247,10 +247,8 @@ def run_test(
247247
trt_inputs = inputs
248248
for num_input in range(num_inputs):
249249
input = inputs[num_input]
250-
if input.dtype in (torch.int64, torch.float64):
251-
dtype_32bit = (
252-
torch.int32 if (input.dtype == torch.int64) else torch.float32
253-
)
250+
if input.dtype in (torch.float64):
251+
dtype_32bit = torch.float32
254252
# should we modify graph here to insert clone nodes?
255253
# ideally not required
256254
trt_inputs = (

0 commit comments

Comments
 (0)