From f65aaa67787e4cb2cd8a9c517da1e3d289c354b4 Mon Sep 17 00:00:00 2001 From: Xuchun Shang Date: Sat, 17 Jan 2026 15:06:07 +0800 Subject: [PATCH] fix for sparse tensor Signed-off-by: Xuchun Shang --- mooncake-integration/store/store_py.cpp | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/mooncake-integration/store/store_py.cpp b/mooncake-integration/store/store_py.cpp index 0d2331755..1ad55e424 100644 --- a/mooncake-integration/store/store_py.cpp +++ b/mooncake-integration/store/store_py.cpp @@ -21,7 +21,7 @@ struct PyTensorInfo { TensorMetadata metadata; // Check validity - bool valid() const { return tensor_size > 0; } + bool valid() const { return true; } }; PyTensorInfo extract_tensor_info(const py::object &tensor, @@ -91,7 +91,7 @@ pybind11::object buffer_to_tensor(BufferHandle *buffer_handle, char *usr_buffer, char *exported_data; if (take_ownership) { total_length = buffer_handle->size(); - if (total_length <= sizeof(TensorMetadata)) { + if (total_length < sizeof(TensorMetadata)) { LOG(ERROR) << "Invalid data format: insufficient data for metadata"; return pybind11::none(); } @@ -110,7 +110,7 @@ pybind11::object buffer_to_tensor(BufferHandle *buffer_handle, char *usr_buffer, return pybind11::none(); } total_length = static_cast(data_length); - if (total_length <= sizeof(TensorMetadata)) { + if (total_length < sizeof(TensorMetadata)) { LOG(ERROR) << "Invalid data format: insufficient data for metadata"; return pybind11::none(); } @@ -129,11 +129,11 @@ pybind11::object buffer_to_tensor(BufferHandle *buffer_handle, char *usr_buffer, TensorDtype dtype_enum = static_cast(metadata.dtype); size_t tensor_size = total_length - sizeof(TensorMetadata); - if (tensor_size == 0 || dtype_enum == TensorDtype::UNKNOWN) { + if (dtype_enum == TensorDtype::UNKNOWN) { if (take_ownership) { delete[] exported_data; } - LOG(ERROR) << "Invalid tensor data or unknown dtype"; + LOG(ERROR) << "Unknown tensor dtype"; return pybind11::none(); }