diff --git a/py/torch_tensorrt/dynamo/runtime/register_fake_class.py b/py/torch_tensorrt/dynamo/runtime/register_fake_class.py index 24777eba88..2d25ddbd4f 100644 --- a/py/torch_tensorrt/dynamo/runtime/register_fake_class.py +++ b/py/torch_tensorrt/dynamo/runtime/register_fake_class.py @@ -4,6 +4,12 @@ import torch +@torch.library.register_fake("tensorrt::execute_engine") +def fake_execute_engine(inputs, trt_engine): + breakpoint() + return trt_engine(inputs) + + # namespace::class_name @torch._library.register_fake_class("tensorrt::Engine") class FakeTRTEngine: