mirror of
https://github.com/kvcache-ai/Mooncake.git
synced 2026-11-05 01:17:55 +00:00
[PR #244] [MERGED] [Feature] support auto-detect topo for both GPUs and CPUs #1424
Labels
No labels
ascend
auto-closed
bug
bug
CI/Build
Common
documentation
enhancement
feature request
good first issue
help wanted
high priority
MetaX
nightly-failure
pull-request
PyTorch Backend
question
RFC
Roadmap
stale
Store
Tests
Transfer Engine
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
kvcache-ai/Mooncake#1424
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
📋 Pull Request Information
Original PR: https://github.com/kvcache-ai/Mooncake/pull/244
Author: @alogfans
Created: 4/14/2025
Status: ✅ Merged
Merged: 4/16/2025
Merged by: @doujiang24
Base:
main← Head:detect-topo📝 Commits (2)
cae6bac[Feature] support auto-detect topo for both GPUs and CPUsaa323e4Merge branch 'main' into detect-topo📊 Changes
10 files changed (+101 additions, -46 deletions)
View changed files
📝
mooncake-integration/transfer_engine/transfer_engine_py.cpp(+51 -30)📝
mooncake-integration/transfer_engine/transfer_engine_py.h(+1 -0)📝
mooncake-transfer-engine/include/memory_location.h(+3 -2)📝
mooncake-transfer-engine/include/transfer_engine.h(+2 -1)📝
mooncake-transfer-engine/src/memory_location.cpp(+32 -3)📝
mooncake-transfer-engine/src/topology.cpp(+3 -4)📝
mooncake-transfer-engine/src/transport/rdma_transport/rdma_transport.cpp(+4 -2)📝
mooncake-transfer-engine/tests/memory_location_test.cpp(+1 -1)📝
mooncake-transfer-engine/tests/rdma_transport_test.cpp(+1 -1)📝
mooncake-transfer-engine/tests/topology_test.cpp(+3 -2)📄 Description
To enable fully auto detection of topology in Transfer Engine, just leave the device name empty in the Python interface.
🔄 This issue represents a GitHub Pull Request. It cannot be merged through Gitea due to API limitations.