[PR #929] [MERGED] feat: add batch_put_from_multi_buffers #1907

New issue

Closed

opened 2026-10-29 21:31:17 +00:00 by bsakaguchi · 0 comments

bsakaguchi commented

2026-10-29 21:31:17 +00:00

Owner

📋 Pull Request Information

Original PR: https://github.com/kvcache-ai/Mooncake/pull/929
Author: @LCAIZJ
Created: 10/15/2025
Status: ✅ Merged
Merged: 10/16/2025
Merged by: @ykwd

Base: main ← Head: dev

📝 Commits (1)

ac8b0c5 add batch_put_from_multi_buffers

📊 Changes

17 files changed (+1072 additions, -178 deletions)

View changed files

📝 docs/source/python-api-reference/mooncake-store.md (+93 -0)
📝 docs/source/python-api-reference/transfer-engine.md (+13 -0)
📝 mooncake-integration/store/store_py.cpp (+82 -20)
📝 mooncake-integration/transfer_engine/transfer_engine_py.cpp (+5 -1)
📝 mooncake-integration/transfer_engine/transfer_engine_py.h (+2 -0)
📝 mooncake-store/include/client.h (+14 -4)
📝 mooncake-store/include/master_service.h (+1 -0)
📝 mooncake-store/include/pybind_client.h (+55 -2)
📝 mooncake-store/include/replica.h (+1 -0)
📝 mooncake-store/include/transfer_task.h (+25 -2)
📝 mooncake-store/src/client.cpp (+308 -98)
📝 mooncake-store/src/master_service.cpp (+3 -1)
📝 mooncake-store/src/pybind_client.cpp (+245 -6)
📝 mooncake-store/src/rpc_service.cpp (+34 -3)
📝 mooncake-store/src/transfer_task.cpp (+99 -41)
📝 mooncake-store/tests/pybind_client_test.cpp (+89 -0)
📝 mooncake-transfer-engine/include/transfer_engine.h (+3 -0)

📄 Description

add batch_put_from_multi_buffers、batch_get_into_multi_buffers to store
add set created transfer engine to store
add prefer_alloc_in_same_node, which aims to achieve best performance for ascend direct transport

_{🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.}

## 📋 Pull Request Information **Original PR:** https://github.com/kvcache-ai/Mooncake/pull/929 **Author:** [@LCAIZJ](https://github.com/LCAIZJ) **Created:** 10/15/2025 **Status:** ✅ Merged **Merged:** 10/16/2025 **Merged by:** [@ykwd](https://github.com/ykwd) **Base:** `main` ← **Head:** `dev` --- ### 📝 Commits (1) - [`ac8b0c5`](https://github.com/kvcache-ai/Mooncake/commit/ac8b0c523dcd346462c7614b4319654e54a603e8) add batch_put_from_multi_buffers ### 📊 Changes **17 files changed** (+1072 additions, -178 deletions) <details> <summary>View changed files</summary> 📝 `docs/source/python-api-reference/mooncake-store.md` (+93 -0) 📝 `docs/source/python-api-reference/transfer-engine.md` (+13 -0) 📝 `mooncake-integration/store/store_py.cpp` (+82 -20) 📝 `mooncake-integration/transfer_engine/transfer_engine_py.cpp` (+5 -1) 📝 `mooncake-integration/transfer_engine/transfer_engine_py.h` (+2 -0) 📝 `mooncake-store/include/client.h` (+14 -4) 📝 `mooncake-store/include/master_service.h` (+1 -0) 📝 `mooncake-store/include/pybind_client.h` (+55 -2) 📝 `mooncake-store/include/replica.h` (+1 -0) 📝 `mooncake-store/include/transfer_task.h` (+25 -2) 📝 `mooncake-store/src/client.cpp` (+308 -98) 📝 `mooncake-store/src/master_service.cpp` (+3 -1) 📝 `mooncake-store/src/pybind_client.cpp` (+245 -6) 📝 `mooncake-store/src/rpc_service.cpp` (+34 -3) 📝 `mooncake-store/src/transfer_task.cpp` (+99 -41) 📝 `mooncake-store/tests/pybind_client_test.cpp` (+89 -0) 📝 `mooncake-transfer-engine/include/transfer_engine.h` (+3 -0) </details> ### 📄 Description 1. add batch_put_from_multi_buffers、batch_get_into_multi_buffers to store 2. add set created transfer engine to store 3. add prefer_alloc_in_same_node, which aims to achieve best performance for ascend direct transport --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>

bsakaguchi

2026-10-29 21:31:17 +00:00