We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5b77d3f commit ce56ca6Copy full SHA for ce56ca6
py/torch_tensorrt/dynamo/conversion/impl/cat.py
@@ -46,7 +46,7 @@ def unify_and_concat_trt_tensors(
46
elif isinstance(x, int) and not has_dynamic and not force_trt_output:
47
t = x # pure static path
48
else:
49
- t = ctx.net.add_constant((1,), np.array([x], dtype=np.int32))
+ t = ctx.net.add_constant((x.numel(),), np.array([x], dtype=np.int32))
50
set_layer_name(t, target, f"{name}_dim{i}_const")
51
t = t.get_output(0)
52
0 commit comments