Skip to content

Commit 83f04ee

Browse files
committed
fix: use dynamo path for conversion utils instead of fx
1 parent 3226b5c commit 83f04ee

File tree

1 file changed

+5
-6
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl/slice

1 file changed

+5
-6
lines changed

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

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
import numpy as np
66
import tensorrt as trt
77
from torch.fx.node import Target
8+
89
from torch_tensorrt.dynamo._SourceIR import SourceIR
910
from torch_tensorrt.dynamo.conversion import impl
1011
from torch_tensorrt.dynamo.conversion._ConversionContext import ConversionContext
@@ -13,6 +14,9 @@
1314
flatten_dims,
1415
get_positive_dim,
1516
get_trt_tensor,
17+
has_dynamic_shape,
18+
prepend_ones,
19+
set_layer_name,
1620
)
1721
from torch_tensorrt.dynamo.conversion.impl.cat import cat
1822
from torch_tensorrt.dynamo.conversion.impl.elementwise import floor_divide
@@ -23,11 +27,6 @@
2327
from torch_tensorrt.dynamo.conversion.impl.shape import shape as get_shape
2428
from torch_tensorrt.dynamo.conversion.impl.slice.base import slice
2529
from torch_tensorrt.dynamo.utils import DYNAMIC_DIM
26-
from torch_tensorrt.fx.converters.converter_utils import (
27-
has_dynamic_shape,
28-
prepend_ones,
29-
set_layer_name,
30-
)
3130
from torch_tensorrt.fx.types import Shape, TRTTensor
3231

3332

@@ -230,7 +229,7 @@ def expand(
230229
# If the rank of the input tensor is less than the shape's rank, pad with ones
231230
if initial_tensor_rank < shape_rank:
232231
input_t = prepend_ones(
233-
ctx.net,
232+
ctx,
234233
input_t,
235234
name + "_expand_broadcast",
236235
shape_rank - initial_tensor_rank,

0 commit comments

Comments
 (0)