tensorrt.py 383 B

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