Skip to content

Commit 015bf19

Browse files
committed
Fixed the CI
1 parent 049b810 commit 015bf19

File tree

1 file changed

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

1 file changed

+9
-10
lines changed

py/torch_tensorrt/dynamo/conversion/impl/select.py

Lines changed: 9 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -258,16 +258,15 @@ def index(
258258
else:
259259
dim_tensor_shape_mult_d1 = transpose_tensor_shape[i]
260260

261-
if isinstance(dim_tensor_shape_mult_d1, TRTTensor):
262-
mult_d1 = convert_binary_elementwise(
263-
ctx,
264-
target,
265-
source_ir,
266-
name + f"_shape_{i}",
267-
trt.ElementWiseOperation.PROD,
268-
mult_d1,
269-
dim_tensor_shape_mult_d1,
270-
)
261+
mult_d1 = convert_binary_elementwise(
262+
ctx,
263+
target,
264+
source_ir,
265+
name + f"_shape_{i}",
266+
trt.ElementWiseOperation.PROD,
267+
mult_d1,
268+
dim_tensor_shape_mult_d1,
269+
)
271270

272271
concat_tensor_layer = ctx.net.add_concatenation(
273272
[

0 commit comments

Comments
 (0)