chore: Move converter registry, update imports

gs-olive · gs-olive · commit b576f309aaaa · 2023-07-25T11:02:18.000-07:00
diff --git a/py/torch_tensorrt/dynamo/__init__.py b/py/torch_tensorrt/dynamo/__init__.py
@@ -5,7 +5,7 @@
     from ._settings import *
     from .conversion import *
     from .aten_tracer import trace
-    from .converter_registry import (
+    from .conversion.converter_registry import (
         DYNAMO_CONVERTERS,
         dynamo_tensorrt_converter,
     )
diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py
@@ -3,7 +3,7 @@
 import torch
 import tensorrt as trt
 from torch_tensorrt.fx.converters import acc_ops_converters
-from ..converter_registry import dynamo_tensorrt_converter
+from .converter_registry import dynamo_tensorrt_converter
 from torch.fx.node import Argument, Target, Node
 
 from torch_tensorrt.fx.types import TRTNetwork, TRTTensor
diff --git a/py/torch_tensorrt/dynamo/conversion/converter_registry.py b/py/torch_tensorrt/dynamo/conversion/converter_registry.py
diff --git a/py/torch_tensorrt/dynamo/conversion/trt_interpreter.py b/py/torch_tensorrt/dynamo/conversion/trt_interpreter.py
@@ -13,7 +13,7 @@
 from torch.fx.node import _get_qualified_name
 from torch.fx.passes.shape_prop import TensorMetadata
 
-from ..converter_registry import DYNAMO_CONVERTERS as CONVERTERS
+from .converter_registry import DYNAMO_CONVERTERS as CONVERTERS
 from torch_tensorrt import Input
 from torch_tensorrt.fx.observer import Observer
 from torch_tensorrt.fx.utils import (

Original file line number	Diff line number	Diff line change
`@@ -5,7 +5,7 @@`
`5`	`5`	`from ._settings import *`
`6`	`6`	`from .conversion import *`
`7`	`7`	`from .aten_tracer import trace`
`8`		`- from .converter_registry import (`
	`8`	`+ from .conversion.converter_registry import (`
`9`	`9`	`DYNAMO_CONVERTERS,`
`10`	`10`	`dynamo_tensorrt_converter,`
`11`	`11`	`)`