correcting the int case

Author: apbose

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

@@ -46,9 +46,17 @@ def unify_and_concat_trt_tensors(
         elif isinstance(x, int) and not has_dynamic and not force_trt_output:
             t = x  # pure static path
         else:
-            t = ctx.net.add_constant((x.numel(),), np.array([x], dtype=np.int32))
-            set_layer_name(t, target, f"{name}_dim{i}_const")
-            t = t.get_output(0)
+            if isinstance(x, int):
+                # wrap int into 1-element np array
+                const_arr = np.array([x], dtype=np.int32)
+                shape = (1,)
+            else:
+                const_arr = np.array(x, dtype=np.int32)
+                shape = (x.numel(),)
+
+        layer = ctx.net.add_constant(shape, const_arr)
+        set_layer_name(layer, target, f"{name}_dim{i}_const")
+        t = layer.get_output(0)
 
         # optional cast
         if cast_dtype and isinstance(t, TRTTensor):