Skip to content

Commit 3710b86

Browse files
committed
removing precision in harness.py interpreter compilationsettings and interpreter.run
1 parent 2ee77fa commit 3710b86

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/py/dynamo/conversion/harness.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -140,9 +140,7 @@ def run_test_custom_compare_results(
140140
if len(expected_ops):
141141
self.assert_has_op(mod, expected_ops)
142142

143-
interpreter_result = interpreter.run(
144-
precision=torch.half if fp16_mode else torch.float
145-
)
143+
interpreter_result = interpreter.run()
146144
trt_mod = PythonTorchTensorRTModule(
147145
interpreter_result.engine,
148146
interpreter_result.input_names,
@@ -288,7 +286,9 @@ def run_test_compare_tensor_attributes_only(
288286
# Previous instance of the interpreter auto-casted 64-bit inputs
289287
# We replicate this behavior here
290288
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,
292292
)
293293

294294
interp = TRTInterpreter(

0 commit comments

Comments
 (0)