chore: updates

pytorch · Oct 28, 2024 · d406366 · d406366
1 parent df13856
commit d406366
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py/torch_tensorrt/dynamo/runtime/register_fake_class.py b/py/torch_tensorrt/dynamo/runtime/register_fake_class.py
@@ -5,8 +5,8 @@
 
 
 @torch.library.register_fake("tensorrt::execute_engine")
-def execute_engine(inputs, trt_engine):
-    breakpoint()
+def fake_tensorrt_execute_engine(inputs, trt_engine):
+    print("==============done==========")
     return trt_engine(inputs)