[V1] fix torch profiling for V1 offline scenarios (#18445)

Signed-off-by: Divakar Verma <divakar.verma@amd.com>
This commit is contained in:
Divakar Verma
2025-05-27 23:16:30 -05:00
committed by GitHub
parent 9a21e331ff
commit 774c5fde30
4 changed files with 23 additions and 51 deletions

View File

@@ -292,6 +292,8 @@ class Worker(WorkerBase):
self.profiler.start()
else:
self.profiler.stop()
print(self.profiler.key_averages().table(
sort_by="self_cuda_time_total"))
def execute_dummy_batch(self) -> None:
self.model_runner._dummy_run(1)