Skip to content

Commit 36c95f8

Browse files
committed
Minor naming bugfix
1 parent 78d4ecd commit 36c95f8

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

py/torch_tensorrt/fx/tracer/dispatch_tracer/tensorrt_dynamo_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def fx2trt(gm: torch.fx.GraphModule, example_inputs, **kwargs):
4949

5050
splitter.node_support_preview()
5151
split_mod = splitter()
52-
num_piece = 0
52+
num_pieces = 0
5353

5454
for name, _ in split_mod.named_children():
5555
print(f"Graph is split into {name}")
@@ -58,7 +58,7 @@ def fx2trt(gm: torch.fx.GraphModule, example_inputs, **kwargs):
5858
# Select threshold above which segmentation is not beneficial and run graph in Torch
5959
if num_pieces > MAX_SPLITS_THRESHOLD:
6060
raise AssertionError(
61-
f"The graph module is split into {num_piece} which is large than the \
61+
f"The graph module is split into {num_pieces} which is large than the \
6262
threshold={MAX_SPLITS_THRESHOLD}. Falling back to non-TRT module."
6363
)
6464

0 commit comments

Comments
 (0)