import torch_tensorrt
from torch_tensorrt.dynamo.runtime._PythonTorchTensorRTModule import (  # noqa: F401
    PythonTorchTensorRTModule,
)
from torch_tensorrt.dynamo.runtime._TorchTensorRTModule import (  # noqa: F401
    TorchTensorRTModule,
)

if torch_tensorrt.ENABLED_FEATURES.torch_tensorrt_runtime:
    from torch_tensorrt.dynamo.runtime.meta_ops.register_meta_ops import *