[PR #1148] [MERGED] [Store] pub_tensor for multiple replica

No labels

No milestone

No project

No assignees

1 participant

Notifications

Due date

The due date is invalid or out of range. Please use the format "yyyy-mm-dd".

No due date set.

Dependencies

No dependencies set.

Reference

kvcache-ai/Mooncake#2039

bsakaguchi commented

2026-10-29 21:32:18 +00:00

Owner

📋 Pull Request Information

Original PR: https://github.com/kvcache-ai/Mooncake/pull/1148
Author: @zxpdemonio
Created: 12/1/2025
Status: ✅ Merged
Merged: 12/8/2025
Merged by: @stmatengss

Base: main ← Head: cruz_zhao/pub_tensor

📝 Commits (10+)

72d1cc6 [Store] pub_tensor for multiple replica
cd7b293 fix
dde11e6 fix
3008f2d fix
52c5512 fix
60fb6d0 [store] support large tensor for pub_tensor() and put_tensor()
1842f33 make RandomAllocationStrategy.Allocate() support perferred_segments
1410bc1 support large tensor
90873a7 [store] fix conflict
227b166 [store] add testcase for pub_tensor(single machine)

📊 Changes

5 files changed (+409 additions, -17 deletions)

View changed files

📝 mooncake-integration/integration_utils.h (+1 -1)
📝 mooncake-integration/store/store_py.cpp (+96 -9)
📝 mooncake-store/include/allocation_strategy.h (+1 -1)
📝 mooncake-store/include/replica.h (+23 -6)
➕ scripts/test_pub_tensor.py (+288 -0)

📄 Description

Description

Type of Change

Types
- Bug fix
- New feature
  - Transfer Engine
  - Mooncake Store
  - Mooncake EP
  - Integration
  - P2P Store
  - Python Wheel
- Breaking change
- CI/CD
- Documentation update
- Other

How Has This Been Tested?

Checklist

I have performed a self-review of my own code.
I have updated the documentation.
I have added tests to prove my changes are effective.

_{🔄 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/1148 **Author:** [@zxpdemonio](https://github.com/zxpdemonio) **Created:** 12/1/2025 **Status:** ✅ Merged **Merged:** 12/8/2025 **Merged by:** [@stmatengss](https://github.com/stmatengss) **Base:** `main` ← **Head:** `cruz_zhao/pub_tensor` --- ### 📝 Commits (10+) - [`72d1cc6`](https://github.com/kvcache-ai/Mooncake/commit/72d1cc656ff2d06f9abe8d89878a0b7370c2ca0d) [Store] pub_tensor for multiple replica - [`cd7b293`](https://github.com/kvcache-ai/Mooncake/commit/cd7b293f9c60f45e842bb8968f90d4f44f863ce3) fix - [`dde11e6`](https://github.com/kvcache-ai/Mooncake/commit/dde11e60ea563e02299e7c9b5758e90ed7c57b1f) fix - [`3008f2d`](https://github.com/kvcache-ai/Mooncake/commit/3008f2d7426d5b15a5ef59aed722f91ecf57529f) fix - [`52c5512`](https://github.com/kvcache-ai/Mooncake/commit/52c5512273caf04142fc502dddedcf38048d4de4) fix - [`60fb6d0`](https://github.com/kvcache-ai/Mooncake/commit/60fb6d0005e82fd78132a2f6af7eba80b4c1fd14) [store] support large tensor for pub_tensor() and put_tensor() - [`1842f33`](https://github.com/kvcache-ai/Mooncake/commit/1842f3303ee97f64d174813a84a4504865e79fcc) make RandomAllocationStrategy.Allocate() support perferred_segments - [`1410bc1`](https://github.com/kvcache-ai/Mooncake/commit/1410bc136c738c18d9f2783dca5d2780396594dc) support large tensor - [`90873a7`](https://github.com/kvcache-ai/Mooncake/commit/90873a7455cb4ed06594d597404b05b65a24e409) [store] fix conflict - [`227b166`](https://github.com/kvcache-ai/Mooncake/commit/227b166f2f6e80f3b0228ba0d136faeabdd5ef27) [store] add testcase for pub_tensor(single machine) ### 📊 Changes **5 files changed** (+409 additions, -17 deletions) <details> <summary>View changed files</summary> 📝 `mooncake-integration/integration_utils.h` (+1 -1) 📝 `mooncake-integration/store/store_py.cpp` (+96 -9) 📝 `mooncake-store/include/allocation_strategy.h` (+1 -1) 📝 `mooncake-store/include/replica.h` (+23 -6) ➕ `scripts/test_pub_tensor.py` (+288 -0) </details> ### 📄 Description ## Description  ## Type of Change * Types - [ ] Bug fix - [ ] New feature - [ ] Transfer Engine - [ ] Mooncake Store - [ ] Mooncake EP - [ ] Integration - [ ] P2P Store - [ ] Python Wheel - [ ] Breaking change - [ ] CI/CD - [ ] Documentation update - [ ] Other ## How Has This Been Tested?  ## Checklist - [ ] I have performed a self-review of my own code. - [ ] I have updated the documentation. - [ ] I have added tests to prove my changes are effective. --- <sub>🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.</sub>

bsakaguchi

closed this issue
added the
pull-request
label

Rows
Columns

[PR #1148] [MERGED] [Store] pub_tensor for multiple replica #2039