tensorrt.py 406 B

1234567891011121314
  1. # mypy: ignore-errors
  2. # import torch # type: ignore[import]
  3. # from .common import device_from_inputs, fake_tensor_unsupported # type: ignore[import]
  4. # from .registry import register_backend # type: ignore[import]
  5. """
  6. Placeholder for TensorRT backend for dynamo via torch-tensorrt
  7. """
  8. # @register_backend
  9. # def tensorrt(gm, example_inputs):
  10. # import torch_tensorrt # type: ignore[import]
  11. # pass