Skip to content

Commit f83615a

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

File tree

1 file changed

+2
-1
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl/activation

1 file changed

+2
-1
lines changed

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

+2-1
Original file line numberDiff line numberDiff line change
@@ -7,14 +7,15 @@
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
1110
from torch_tensorrt.fx.converters.converter_utils import (
1211
get_trt_plugin,
1312
mark_as_int8_layer,
1413
set_layer_name,
1514
)
1615
from torch_tensorrt.fx.types import TRTNetwork, TRTPluginFieldCollection, TRTTensor
1716

17+
from .base import convert_activation
18+
1819

1920
def gelu(
2021
network: TRTNetwork,

0 commit comments

Comments
 (0)