Skip to content

Commit 4e0288a

Browse files
committed
fix bugs: circular import
1 parent cd7e98d commit 4e0288a

File tree

5 files changed

+11
-11
lines changed

5 files changed

+11
-11
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

+9-9
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def aten_ops_gelu(
160160
kwargs: Dict[str, Argument],
161161
name: str,
162162
) -> Union[TRTTensor, Sequence[TRTTensor]]:
163-
return impl.activation.gelu(
163+
return impl.actv.gelu(
164164
network,
165165
target,
166166
SourceIR.ATEN,
@@ -177,7 +177,7 @@ def aten_ops_relu(
177177
kwargs: Dict[str, Argument],
178178
name: str,
179179
) -> Union[TRTTensor, Sequence[TRTTensor]]:
180-
return impl.activation.relu(
180+
return impl.actv.relu(
181181
network,
182182
target,
183183
SourceIR.ATEN,
@@ -194,7 +194,7 @@ def aten_ops_sigmoid(
194194
kwargs: Dict[str, Argument],
195195
name: str,
196196
) -> Union[TRTTensor, Sequence[TRTTensor]]:
197-
return impl.activation.sigmoid(
197+
return impl.actv.sigmoid(
198198
network,
199199
target,
200200
SourceIR.ATEN,
@@ -211,7 +211,7 @@ def aten_ops_tanh(
211211
kwargs: Dict[str, Argument],
212212
name: str,
213213
) -> Union[TRTTensor, Sequence[TRTTensor]]:
214-
return impl.activation.tanh(
214+
return impl.actv.tanh(
215215
network,
216216
target,
217217
SourceIR.ATEN,
@@ -228,7 +228,7 @@ def aten_ops_leaky_relu(
228228
kwargs: Dict[str, Argument],
229229
name: str,
230230
) -> Union[TRTTensor, Sequence[TRTTensor]]:
231-
return impl.activation.leaky_relu(
231+
return impl.actv.leaky_relu(
232232
network,
233233
target,
234234
SourceIR.ATEN,
@@ -246,7 +246,7 @@ def aten_ops_elu(
246246
kwargs: Dict[str, Argument],
247247
name: str,
248248
) -> Union[TRTTensor, Sequence[TRTTensor]]:
249-
return impl.activation.elu(
249+
return impl.actv.elu(
250250
network,
251251
target,
252252
SourceIR.ATEN,
@@ -265,7 +265,7 @@ def aten_ops_softplus(
265265
kwargs: Dict[str, Argument],
266266
name: str,
267267
) -> Union[TRTTensor, Sequence[TRTTensor]]:
268-
return impl.activation.softplus(
268+
return impl.actv.softplus(
269269
network,
270270
target,
271271
SourceIR.ATEN,
@@ -283,7 +283,7 @@ def aten_ops_clip(
283283
kwargs: Dict[str, Argument],
284284
name: str,
285285
) -> Union[TRTTensor, Sequence[TRTTensor]]:
286-
return impl.activation.clip(
286+
return impl.actv.clip(
287287
network,
288288
target,
289289
SourceIR.ATEN,
@@ -302,7 +302,7 @@ def aten_ops_hard_sigmoid(
302302
kwargs: Dict[str, Argument],
303303
name: str,
304304
) -> Union[TRTTensor, Sequence[TRTTensor]]:
305-
return impl.activation.hard_sigmoid(
305+
return impl.actv.hard_sigmoid(
306306
network,
307307
target,
308308
SourceIR.ATEN,

py/torch_tensorrt/dynamo/conversion/impl/__init__.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
from torch_tensorrt.fx.converters.impl import convolution
22

33
from . import (
4-
activation,
4+
actv,
55
cast,
66
condition,
77
elementwise,

py/torch_tensorrt/dynamo/conversion/impl/activation/ops.py renamed to py/torch_tensorrt/dynamo/conversion/impl/actv/ops.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
from torch import Tensor
88
from torch.fx.node import Target
99
from torch_tensorrt.dynamo._SourceIR import SourceIR
10-
from torch_tensorrt.dynamo.conversion.impl.activation.base import convert_activation
10+
from torch_tensorrt.dynamo.conversion.impl.actv.base import convert_activation
1111
from torch_tensorrt.fx.converters.converter_utils import (
1212
get_trt_plugin,
1313
mark_as_int8_layer,

0 commit comments

Comments
 (0)