Skip to content

Commit d4a83ac

Browse files
committed
fix type bug
1 parent 58c8f2f commit d4a83ac

File tree

1 file changed

+1
-9
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl/slice

1 file changed

+1
-9
lines changed

py/torch_tensorrt/dynamo/conversion/impl/slice/ops.py

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -180,15 +180,7 @@ def tile(
180180
# prepend 1 to dims
181181
dims = (1,) * -diff + tuple(dims)
182182

183-
if all(isinstance(d, int) for d in dims):
184-
shapes = [i * j for i, j in zip(input.shape, dims)]
185-
else:
186-
shapes = []
187-
for i, (s, d) in enumerate(zip(input.shape, dims)):
188-
shapes.append(
189-
impl.elementwise.mul(ctx, target, source_ir, f"{name}_mul_{i}", s, d)
190-
)
191-
183+
shapes = [i * j for i, j in zip(input.shape, dims)]
192184
starts = [0] * len(dims)
193185
strides = [1] * len(dims)
194186
layer = ctx.net.add_slice(input, tuple(starts), tuple(shapes), tuple(strides))

0 commit comments

Comments
 (0)