refine stdout logging codes (#13015)
This commit is contained in:
@@ -1968,7 +1968,7 @@ class Scheduler(
|
||||
if batch.forward_mode == ForwardMode.EXTEND:
|
||||
current_time = time.perf_counter()
|
||||
for req in batch.reqs:
|
||||
req.time_stats.prefill_start_time = current_time
|
||||
req.time_stats.prefill_start_time_host = current_time
|
||||
|
||||
# Place holder handling for pd-disagg decode event loop
|
||||
if batch.forward_mode.is_prebuilt():
|
||||
@@ -2077,7 +2077,7 @@ class Scheduler(
|
||||
if batch.forward_mode == ForwardMode.EXTEND:
|
||||
current_time = time.perf_counter()
|
||||
for req in batch.reqs:
|
||||
req.time_stats.prefill_end_time = current_time
|
||||
req.time_stats.prefill_end_time_host = current_time
|
||||
|
||||
return ret
|
||||
|
||||
|
||||
Reference in New Issue
Block a user