File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
tests/py/dynamo/conversion Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -140,9 +140,7 @@ def run_test_custom_compare_results(
140
140
if len (expected_ops ):
141
141
self .assert_has_op (mod , expected_ops )
142
142
143
- interpreter_result = interpreter .run (
144
- precision = torch .half if fp16_mode else torch .float
145
- )
143
+ interpreter_result = interpreter .run ()
146
144
trt_mod = PythonTorchTensorRTModule (
147
145
interpreter_result .engine ,
148
146
interpreter_result .input_names ,
@@ -288,7 +286,9 @@ def run_test_compare_tensor_attributes_only(
288
286
# Previous instance of the interpreter auto-casted 64-bit inputs
289
287
# We replicate this behavior here
290
288
compilation_settings = CompilationSettings (
291
- precision = precision , truncate_long_and_double = True
289
+ enabled_precisions = {dtype ._from (precision )},
290
+ truncate_long_and_double = True ,
291
+ debug = True ,
292
292
)
293
293
294
294
interp = TRTInterpreter (
You can’t perform that action at this time.
0 commit comments