add impl.activation in setup to resolve cyclic import issue

pytorch · Aug 31, 2023 · a4722d9 · a4722d9
1 parent 6296abc
commit a4722d9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 11 deletions.
diff --git a/py/torch_tensorrt/dynamo/conversion/impl/activation/ops.py b/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,

diff --git a/setup.py b/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",