File tree Expand file tree Collapse file tree 2 files changed +17
-6
lines changed Expand file tree Collapse file tree 2 files changed +17
-6
lines changed Original file line number Diff line number Diff line change @@ -115,9 +115,4 @@ def _call(*initial_args):
115115 binding .copy_outputs_to_cpu ()
116116 return outputs
117117
118- return _call
119-
120-
121- @register_backend
122- def tensorrt (gm , example_inputs ):
123- return onnxrt (gm , example_inputs , provider = "TensorrtExecutionProvider" )
118+ return _call
Original file line number Diff line number Diff line change 1+ import importlib
2+ import os
3+ import tempfile
4+
5+ import torch
6+ from .common import device_from_inputs , fake_tensor_unsupported
7+ from .registry import register_backend
8+
9+ '''
10+ Placeholder for TensorRT backend for dynamo via torch-tensorrt
11+ '''
12+
13+ # @register_backend
14+ # def tensorrt(gm, example_inputs):
15+ # import torch_tensorrt # type: ignore[import]
16+ # pass
You can’t perform that action at this time.
0 commit comments