fix bugs: circular import

pytorch · Aug 23, 2023 · f83615a · f83615a
1 parent cd7e98d
commit f83615a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/py/torch_tensorrt/dynamo/conversion/impl/activation/ops.py b/py/torch_tensorrt/dynamo/conversion/impl/activation/ops.py
@@ -7,14 +7,15 @@
 from torch import Tensor
 from torch.fx.node import Target
 from torch_tensorrt.dynamo._SourceIR import SourceIR
-from torch_tensorrt.dynamo.conversion.impl.activation.base import convert_activation
 from torch_tensorrt.fx.converters.converter_utils import (
     get_trt_plugin,
     mark_as_int8_layer,
     set_layer_name,
 )
 from torch_tensorrt.fx.types import TRTNetwork, TRTPluginFieldCollection, TRTTensor
 
+from .base import convert_activation
+
 
 def gelu(
     network: TRTNetwork,