microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-25 23:34:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Configure IPC domain per communicator

Browse Source 
Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
This commit is contained in:
Binyang Li
2026-05-15 22:26:53 +00:00
parent ee82cc4c41
commit dbebde2b58
10 changed files with 41 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/csrc/core_py.cpp
View File

@@ -282,6 +282,8 @@ void register_core(nb::module_& m) {
nb::arg("context") = nullptr)
.def("bootstrap", &Communicator::bootstrap)
.def("context", &Communicator::context)
.def("set_ipc_domain_n_ranks", &Communicator::setIpcDomainNranks, nb::arg("n_ranks"))
.def("get_ipc_domain_n_ranks", &Communicator::getIpcDomainNranks)
.def(
"register_memory",
[](Communicator* self, uintptr_t ptr, size_t size, TransportFlags transports) {