From 1aefbef0b3f52ae069048db5c1c5ab480abd8882 Mon Sep 17 00:00:00 2001 From: qwes5s5 <45442318+qwes5s5@users.noreply.github.com> Date: Fri, 5 Dec 2025 14:45:52 +0800 Subject: [PATCH] fix trace log (#5386) --- fastdeploy/engine/common_engine.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/fastdeploy/engine/common_engine.py b/fastdeploy/engine/common_engine.py index cc02363c4..51419bde9 100644 --- a/fastdeploy/engine/common_engine.py +++ b/fastdeploy/engine/common_engine.py @@ -862,9 +862,14 @@ class EngineService: ) self.resource_manager.get_real_bsz() for task in tasks: - trace_print(LoggingEventName.RESOURCE_ALLOCATE_END, task.request_id, getattr(task, "user", "")) - trace_print(LoggingEventName.REQUEST_SCHEDULE_END, task.request_id, getattr(task, "user", "")) - trace_print(LoggingEventName.INFERENCE_START, task.request_id, getattr(task, "user", "")) + if task.task_type == RequestType.PREFILL: + trace_print( + LoggingEventName.RESOURCE_ALLOCATE_END, task.request_id, getattr(task, "user", "") + ) + trace_print( + LoggingEventName.REQUEST_SCHEDULE_END, task.request_id, getattr(task, "user", "") + ) + trace_print(LoggingEventName.INFERENCE_START, task.request_id, getattr(task, "user", "")) self.engine_worker_queue.put_tasks((tasks, self.resource_manager.real_bsz)) # 4. Response error tasks