fix: Register tensorrt backend name (#2311)

Author: gs-olive

py/torch_tensorrt/dynamo/backend/backends.py

@@ -31,6 +31,7 @@
 logger = logging.getLogger(__name__)
 
 
+@td.register_backend(name="tensorrt")  # type: ignore[misc]
 @td.register_backend(name="torch_tensorrt")  # type: ignore[misc]
 def torch_tensorrt_backend(
     gm: torch.fx.GraphModule, sample_inputs: Sequence[torch.Tensor], **kwargs: Any

tests/py/dynamo/backend/test_backend_compiler.py

@@ -312,5 +312,13 @@ def forward(self, x, y):
         )
 
 
+class TestRegistration(TestCase):
+    def test_torch_tensorrt_registration(self):
+        self.assertIn("torch_tensorrt", torch._dynamo.list_backends())
+
+    def test_tensorrt_registration(self):
+        self.assertIn("tensorrt", torch._dynamo.list_backends())
+
+
 if __name__ == "__main__":
     run_tests()