From bbecc5365dfca0dff686c9805dca2e12571c1b67 Mon Sep 17 00:00:00 2001 From: Wei Date: Tue, 30 Aug 2022 13:52:50 -0700 Subject: [PATCH] Update lower.py --- py/torch_tensorrt/fx/lower.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/py/torch_tensorrt/fx/lower.py b/py/torch_tensorrt/fx/lower.py index 59b59d580f..9a641e000e 100644 --- a/py/torch_tensorrt/fx/lower.py +++ b/py/torch_tensorrt/fx/lower.py @@ -86,7 +86,9 @@ def create(cls, lower_setting): def __call__(self, mod, input, split_name) -> TRTInterpreterResult: assert self.lower_setting.input_specs, "Can't find input specs for lowering!" - logger.info(f"{split_name=} {self.lower_setting.input_specs=}") + logger.info( + f"split_name={split_name}, input_specs={self.lower_setting.input_specs}" + ) # Prepare algorithm selector and timing_cache for TRTInterpreter algo_selector = None