microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-04-19 22:39:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make IB more configurable (#703)

Browse Source 
* Added `port` and `gidIndex` field in the IB endpoint config (and
`deviceIndex` field for future usages)
* Added `MSCCLPP_IBV_SO` env variable to specify a custom libibverbs.so
* Added `--ib_gid_index` CLI option to `mp_unit_tests`
* Other minor fixes
This commit is contained in:
Changho Hwang
2025-12-18 13:21:07 -08:00
committed by GitHub
parent 11b7b35832
commit 9e076da3d4
14 changed files with 182 additions and 84 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/test/_cpp/proxy_test.cpp
View File

@@ -31,12 +31,12 @@ class MyProxyService {
proxy_([&](mscclpp::ProxyTrigger triggerRaw) { return handleTrigger(triggerRaw); }) {
allRegMem_.reserve(allRegMemList.size());
for (size_t i = 0; i < allRegMemList.size(); ++i) {
auto& regMem = nb::cast<const mscclpp::RegisteredMemory&>(allRegMemList[i]);
allRegMem_.push_back(regMem);
auto regMem = nb::cast<mscclpp::RegisteredMemory>(allRegMemList[i]);
allRegMem_.emplace_back(regMem);
}
semaphores_.reserve(semaphoreList.size());
for (size_t i = 0; i < semaphoreList.size(); ++i) {
auto& sema = nb::cast<const mscclpp::Semaphore&>(semaphoreList[i]);
auto sema = nb::cast<mscclpp::Semaphore>(semaphoreList[i]);
semaphores_.emplace_back(sema);
}
}