Update lower.py (#1324)

pytorch · Aug 30, 2022 · e374eb1 · e374eb1
1 parent 81f2a5c
commit e374eb1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git 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