Merge pull request #1104 from pytorch/fb-sync-2-wwei6

Refactor the internal codebase from fx2trt_oss to torch_tensorrt

Author: Wei

docs/_modules/torch_tensorrt/_compile.html

@@ -518,7 +518,7 @@ <h1>Source code for torch_tensorrt._compile</h1><div class="highlight"><pre>
                     <span class="c1"># profiling_verbosity=trt.ProfilingVerbosity.DETAILED, #For profile</span>
                 <span class="p">)</span>
                 <span class="c1"># For profile</span>
-                <span class="c1"># from fx2trt_oss.fx.tools.trt_profiler_sorted import profile_trt_module</span>
+                <span class="c1"># from torch_tensorrt.fx.tools.trt_profiler_sorted import profile_trt_module</span>
                 <span class="c1"># profile_trt_module(&quot;&quot;, trt_mod, acc_inputs)</span>
                 <span class="n">trt_mod</span> <span class="o">=</span> <span class="n">TRTModule</span><span class="p">(</span><span class="o">*</span><span class="n">r</span><span class="p">)</span>
 

py/torch_tensorrt/_compile.py

@@ -172,7 +172,7 @@ def get_input(self, inputs):
                     # profiling_verbosity=trt.ProfilingVerbosity.DETAILED, #For profile
                 )
                 # For profile
-                # from fx2trt_oss.fx.tools.trt_profiler_sorted import profile_trt_module
+                # from torch_tensorrt.fx.tools.trt_profiler_sorted import profile_trt_module
                 # profile_trt_module("", trt_mod, acc_inputs)
                 trt_mod = TRTModule(*r)
 

py/torch_tensorrt/fx/converters/acc_ops_converters.py

@@ -4,22 +4,21 @@
 import warnings
 from typing import cast, Dict, Optional, Sequence, Tuple, Union
 
-from ..tracer.acc_tracer import acc_ops
 import numpy as np
 
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
+
+from ..tracer.acc_tracer import acc_ops
 from ..types import *  # noqa: F403
-from ..utils import (
-    get_dynamic_dims,
-    torch_dtype_from_trt,
-    torch_dtype_to_trt,
-)
 from torch.fx.immutable_collections import immutable_list
 from torch.fx.node import Argument, Target
 
+from ..utils import get_dynamic_dims, torch_dtype_from_trt, torch_dtype_to_trt
+
 from .converter_utils import *  # noqa: F403
 
 

py/torch_tensorrt/fx/converters/activation.py

@@ -3,6 +3,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import mark_as_int8_layer

py/torch_tensorrt/fx/converters/adaptive_avgpool.py

@@ -1,6 +1,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import extend_mod_attr_to_tuple, mark_as_int8_layer

py/torch_tensorrt/fx/converters/add.py

@@ -3,6 +3,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import get_dyn_range, mark_as_int8_layer

py/torch_tensorrt/fx/converters/batchnorm.py

@@ -3,6 +3,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import get_dyn_range, mark_as_int8_layer, to_numpy

py/torch_tensorrt/fx/converters/converter_utils.py

@@ -7,6 +7,8 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+from torch.fx.node import Argument, Target
+
 from ..types import (
     Shape,
     TRTDataType,
@@ -18,7 +20,6 @@
     TRTTensor,
 )
 from ..utils import torch_dtype_from_trt
-from torch.fx.node import Argument, Target
 
 
 def get_trt_plugin(

py/torch_tensorrt/fx/converters/convolution.py

@@ -2,6 +2,7 @@
 import numpy as np
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import (

py/torch_tensorrt/fx/converters/linear.py

@@ -1,6 +1,7 @@
 # @manual=//deeplearning/trt/python:py_tensorrt
 import tensorrt as trt
 import torch
+
 from ..converter_registry import tensorrt_converter
 
 from .converter_utils import get_dyn_range, mark_as_int8_layer, to_numpy