microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-12 09:17:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'main' into olli/autosetup-v2

Browse Source
This commit is contained in:
Changho Hwang
2023-11-14 18:00:43 +08:00
committed by GitHub
parent 2afce036ad 3521fb0280
commit d12dd10488
8 changed files with 119 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -39,7 +39,7 @@ class MyProxyService {
semaphores_(semaphores),
proxy_([&](mscclpp::ProxyTrigger triggerRaw) { return handleTrigger(triggerRaw); }, [&]() { bindThread(); }) {
int cudaDevice;
cudaGetDevice(&cudaDevice);
MSCCLPP_CUDATHROW(cudaGetDevice(&cudaDevice));
deviceNumaNode_ = mscclpp::getDeviceNumaNode(cudaDevice);
}