From 31165b857e511e38e8c9b2123252b52eec2fafbe Mon Sep 17 00:00:00 2001 From: Evan Li Date: Tue, 16 Jul 2024 13:09:01 -0700 Subject: [PATCH] revert backend changes --- py/torch_tensorrt/dynamo/backend/backends.py | 20 ++++++++++---------- 1 file changed, 10 insertions(+), 10 deletions(-) diff --git a/py/torch_tensorrt/dynamo/backend/backends.py b/py/torch_tensorrt/dynamo/backend/backends.py index 7cef6b0e67..cfb1fed3b3 100644 --- a/py/torch_tensorrt/dynamo/backend/backends.py +++ b/py/torch_tensorrt/dynamo/backend/backends.py @@ -96,19 +96,19 @@ def _pretraced_backend( ), ) - logger.debug("Post-AOT Autograd graph:\n" + str(gm.graph)) + logger.debug("Post-AOT Autograd graph:\n" + str(gm.graph)) - gm = post_lowering(gm, sample_inputs) + gm = post_lowering(gm, sample_inputs) - logger.debug("Lowered Input graph:\n " + str(gm.graph)) + logger.debug("Lowered Input graph:\n " + str(gm.graph)) - torchtrt_inputs = prepare_inputs(torch_inputs, disable_memory_format_check=True) - trt_compiled = compile_module( - gm, - torchtrt_inputs, - settings=settings, - ) - return trt_compiled + torchtrt_inputs = prepare_inputs(torch_inputs, disable_memory_format_check=True) + trt_compiled = compile_module( + gm, + torchtrt_inputs, + settings=settings, + ) + return trt_compiled except (AssertionError, RuntimeError): if not settings.pass_through_build_failures: logger.warning(