MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT-LLM / _send_data_async

Method _send_data_async

tensorrt_llm/executor/ipc.py:358–368  ·  view source on GitHub ↗

Async version of _send_data.

(self,
                               data: bytes,
                               routing_id: Optional[bytes] = None)

Source from the content-addressed store, hash-verified

356 self.socket.send(data, flags=flags)
357
358 async def _send_data_async(self,
359 data: bytes,
360 routing_id: Optional[bytes] = None):
361 """Async version of _send_data."""
362 if self.socket_type == zmq.ROUTER:
363 identity = routing_id if routing_id is not None else self._last_identity
364 if identity is None:
365 raise ValueError("ROUTER socket requires identity")
366 await self.socket.send_multipart([identity, data])
367 else:
368 await self.socket.send(data)
369
370 def _recv_data(self, return_identity: bool = False) -> Any:
371 """Receive data using appropriate API based on socket type."""

Callers 1

put_asyncMethod · 0.95

Calls 1

sendMethod · 0.45

Tested by

no test coverage detected