to revert: debug statements for CI

Signed-off-by: alec-flowers <[email protected]>

Author: alec-flowers

tests/v1/engine/test_engine_core_client.py

@@ -331,12 +331,16 @@ def test_kv_cache_events(
                 "Token ids should be the same as the custom tokens")
             assert event.token_ids == custom_tokens, (
                 "Token ids should be the same as the custom tokens")
+            print("Asserts Finished")
         finally:
             client.shutdown()
+            print("Client Shutdown")
             subscriber.close()
+            print("Subscriber Shutdown")
             # TODO hack to try and fix CI hang
             ctx = zmq.Context.instance()
             ctx.term()
+            print("Context Shutdown")
         return
 
 

vllm/distributed/kv_events.py

@@ -171,6 +171,7 @@ def shutdown(self) -> None:
             self._thread.join(timeout=self.SHUTDOWN_TIMEOUT)
 
         # Clean up ZMQ resources
+        logger.info("ZmqEventPublisher - Shutting down ZMQ sockets")
         try:
             if self._pub is not None:
                 self._pub.close(linger=0)

vllm/v1/core/sched/scheduler.py

@@ -854,4 +854,5 @@ def make_spec_decoding_stats(
 
     def shutdown(self) -> None:
         if self.kv_event_publisher:
+            logger.info("scheduler.py - Shutting down kv event publisher")
             self.kv_event_publisher.shutdown()

vllm/v1/engine/core.py

@@ -258,6 +258,7 @@ def shutdown(self):
         if self.model_executor:
             self.model_executor.shutdown()
         if self.scheduler:
+            logger.info("core.py - Shutting down scheduler")
             self.scheduler.shutdown()
 
     def profile(self, is_start: bool = True):

vllm/v1/engine/core_client.py

@@ -323,6 +323,8 @@ def __call__(self):
 
         # ZMQ context termination can hang if the sockets
         # aren't explicitly closed first.
+
+        logger.info("BackgroundResources - Closing output socket")
         if self.output_socket is not None:
             self.output_socket.close(linger=0)
         if self.input_socket is not None:
@@ -334,6 +336,7 @@ def __call__(self):
                 shutdown_sender.connect(self.shutdown_path)
                 # Send shutdown signal.
                 shutdown_sender.send(b'')
+                logger.info("BackgroundResources - Shutdown signal sent")
 
     def validate_alive(self, frames: Sequence[zmq.Frame]):
         if len(frames) == 1 and (frames[0].buffer