File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
tests/py/dynamo/conversion Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -249,7 +249,7 @@ def run_test(
249
249
input = inputs [num_input ]
250
250
if input .dtype in (torch .int64 , torch .float64 ):
251
251
dtype_32bit = (
252
- torch .int32 if (input .dtype == torch .int64 ) else torch .int64
252
+ torch .int32 if (input .dtype == torch .int64 ) else torch .float32
253
253
)
254
254
# should we modify graph here to insert clone nodes?
255
255
# ideally not required
@@ -259,7 +259,7 @@ def run_test(
259
259
input .to (dtype_32bit ),
260
260
]
261
261
+ list (trt_inputs [num_input + 1 :])
262
- )
262
+ )
263
263
264
264
trt_input_specs = [Input .from_tensor (i ) for i in trt_inputs ]
265
265
input_specs = [Input .from_tensor (i ) for i in inputs ]
@@ -270,7 +270,7 @@ def run_test(
270
270
mod ,
271
271
input_specs ,
272
272
compilation_settings .device ,
273
- truncate_long_and_double = compilation_settings .truncate_long_and_double ,
273
+ truncate_double = compilation_settings .truncate_double ,
274
274
)
275
275
276
276
_LOGGER .debug (f"Compilation settings: { compilation_settings } " )
You can’t perform that action at this time.
0 commit comments