Skip to content

Commit fed0533

Browse files
committed
fix a concat bug
1 parent 85e28ac commit fed0533

File tree

1 file changed

+2
-2
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl

1 file changed

+2
-2
lines changed

py/torch_tensorrt/dynamo/conversion/impl/cat.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,10 +24,10 @@ def cat(
2424
trt_inputs = []
2525
for i, each_input in enumerate(input):
2626
if not isinstance(each_input, TRTTensor):
27-
each_input = get_trt_tensor(ctx, each_input, name + f"_tensor_{i}")
27+
each_input = get_trt_tensor(ctx, each_input, f"{name}_tensor_{i}")
2828
trt_inputs.append(each_input)
2929
concat_layer = ctx.net.add_concatenation(trt_inputs)
3030
dim = get_positive_dim(dim, len(input[0].shape))
3131
concat_layer.axis = dim
32-
set_layer_name(concat_layer, target, name + "_gather", source_ir)
32+
set_layer_name(concat_layer, target, f"{name}_gather", source_ir)
3333
return concat_layer.get_output(0)

0 commit comments

Comments
 (0)