Skip to content

Commit 087f97d

Browse files
author
Wei
committed
Update lower.py (#1324)
1 parent 4ed2aac commit 087f97d

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

py/torch_tensorrt/fx/lower.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,9 @@ def create(cls, lower_setting):
8686

8787
def __call__(self, mod, input, split_name) -> TRTInterpreterResult:
8888
assert self.lower_setting.input_specs, "Can't find input specs for lowering!"
89-
logger.info(f"{split_name=} {self.lower_setting.input_specs=}")
89+
logger.info(
90+
f"split_name={split_name}, input_specs={self.lower_setting.input_specs}"
91+
)
9092

9193
# Prepare algorithm selector and timing_cache for TRTInterpreter
9294
algo_selector = None

0 commit comments

Comments
 (0)