add impl.activation in setup to resolve cyclic import issue

Author: zewenli98

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

@@ -5,10 +5,9 @@
 import torch
 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.types import TRTNetwork, TRTTensor
 
-from .base import convert_activation
-
 
 def relu(
     network: TRTNetwork,

setup.py

@@ -1,12 +1,16 @@
 import glob
 import os
 import platform
+import re
 import subprocess
 import sys
 import warnings
 from dataclasses import dataclass
+from datetime import datetime
 from distutils.cmd import Command
+from pathlib import Path
 from shutil import copyfile, rmtree
+from typing import List
 
 import setuptools
 import yaml
@@ -18,15 +22,6 @@
 from torch.utils import cpp_extension
 from wheel.bdist_wheel import bdist_wheel
 
-import yaml
-import re
-import os
-import subprocess
-
-from datetime import datetime
-from pathlib import Path
-from typing import List
-
 __version__: str = "0.0.0"
 __cuda_version__: str = "0.0"
 __cudnn_version__: str = "0.0"
@@ -389,6 +384,7 @@ def run(self):
     "torch_tensorrt.dynamo.backend",
     "torch_tensorrt.dynamo.conversion",
     "torch_tensorrt.dynamo.conversion.impl",
+    "torch_tensorrt.dynamo.conversion.impl.activation",
     "torch_tensorrt.dynamo.conversion.impl.condition",
     "torch_tensorrt.dynamo.conversion.impl.elementwise",
     "torch_tensorrt.dynamo.conversion.impl.normalization",
@@ -415,6 +411,7 @@ def run(self):
     "torch_tensorrt.dynamo.backend": "py/torch_tensorrt/dynamo/backend",
     "torch_tensorrt.dynamo.conversion": "py/torch_tensorrt/dynamo/conversion",
     "torch_tensorrt.dynamo.conversion.impl": "py/torch_tensorrt/dynamo/conversion/impl",
+    "torch_tensorrt.dynamo.conversion.impl.activation": "py/torch_tensorrt/dynamo/conversion/impl/activation",
     "torch_tensorrt.dynamo.conversion.impl.condition": "py/torch_tensorrt/dynamo/conversion/impl/condition",
     "torch_tensorrt.dynamo.conversion.impl.elementwise": "py/torch_tensorrt/dynamo/conversion/impl/elementwise",
     "torch_tensorrt.dynamo.conversion.impl.normalization": "py/torch_tensorrt/dynamo/conversion/impl/normalization",