chore: updates

pytorch · Oct 23, 2024 · fca16a5 · fca16a5
1 parent 839c72e
commit fca16a5
Showing 1 changed file with 6 additions and 0 deletions.
diff --git 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: