Improve profiler and integrate profiler in bench_one_batch_server (#6787)

2025-05-31 15:53:55 -07:00
parent b520d02888
commit 2d72fc47cf
25 changed files with 481 additions and 223 deletions
@@ -350,6 +350,7 @@ async def start_profile_async(obj: Optional[ProfileReqInput] = None):
        activities=obj.activities,
        with_stack=obj.with_stack,
        record_shapes=obj.record_shapes,
+        profile_by_stage=obj.profile_by_stage,
    )
    return Response(
        content="Start profiling.\n",