We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6afbb8d commit 9ca3f62Copy full SHA for 9ca3f62
tests/py/dynamo/conversion/harness.py
@@ -247,10 +247,8 @@ def run_test(
247
trt_inputs = inputs
248
for num_input in range(num_inputs):
249
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
- )
+ if input.dtype in (torch.float64):
+ dtype_32bit = torch.float32
254
# should we modify graph here to insert clone nodes?
255
# ideally not required
256
trt_inputs = (
0 commit comments