Reserve the tensorrt backend name for torch-tensorrt (#95627)

Wei · web-flow · commit c4fa85082781 · 2023-02-27T17:17:47.000-08:00
diff --git a/torch/_dynamo/backends/onnxrt.py b/torch/_dynamo/backends/onnxrt.py
@@ -116,8 +116,3 @@ def _call(*initial_args):
         return outputs
 
     return _call
-
-
-@register_backend
-def tensorrt(gm, example_inputs):
-    return onnxrt(gm, example_inputs, provider="TensorrtExecutionProvider")
diff --git a/torch/_dynamo/backends/tensorrt.py b/torch/_dynamo/backends/tensorrt.py
@@ -0,0 +1,12 @@
+# import torch  # type: ignore[import]
+# from .common import device_from_inputs, fake_tensor_unsupported  # type: ignore[import]
+# from .registry import register_backend  # type: ignore[import]
+
+"""
+Placeholder for TensorRT backend for dynamo via torch-tensorrt
+"""
+
+# @register_backend
+# def tensorrt(gm, example_inputs):
+#    import torch_tensorrt # type: ignore[import]
+#    pass
