o پi@sHddlZddlmZmZddlZGdddeZGdddejdZdS)N)ABCabstractmethodc@s$eZdZedejdefddZdS) SerializertreturncCt)aZ Serialize a pytorch tensor to bytes. The serialized bytes should contain both the data and the metadata (shape, dtype, etc.) of the tensor. Input: t: the input pytorch tensor, can be on any device, in any shape, with any dtype Returns: bytes: the serialized bytes NotImplementedError)selfrr T/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/connector/serde/serde.pyto_bytes s zSerializer.to_bytesN)__name__ __module__ __qualname__rtorchTensorbytesr r r r r r src@s,eZdZddZededejfddZdS) DeserializercCs ||_dS)N)dtype)r rr r r __init__s zDeserializer.__init__bsrcCr)z Deserialize a pytorch tensor from bytes. Input: bytes: a stream of bytes Output: torch.Tensor: the deserialized pytorch tensor r)r rr r r from_bytes s zDeserializer.from_bytesN) rrrrrrrrrr r r r rsr) metaclass)abcrrrrABCMetarr r r r s