yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add NCCL_NET flag to many debug lines.

Browse Source
This commit is contained in:
Sylvain Jeaugey 2018-12-04 13:10:19 -08:00
parent cdae05b277
commit b8a9a32ccb
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/init.cu
View File

@ -77,7 +77,7 @@ ncclResult_t initNet(ncclNet_t* net) {
NCCLCHECK(net->init(ncclDebugLog));
NCCLCHECK(net->devices(&ndev));
if (ndev <= 0) {
INFO(NCCL_INIT, "Net/%s: call to devices() returned 0 devices.", net->name);
INFO(NCCL_INIT|NCCL_NET, "Net/%s: call to devices() returned 0 devices.", net->name);
return ncclSystemError;
}
return ncclSuccess;
@ -90,15 +90,15 @@ ncclResult_t initNetPlugin(ncclNet_t** net) {
// string, so checking errno doesn't hurt to try to provide a better
// error message
if (errno == ENOENT) {
INFO(NCCL_INIT, "No network plugin found.");
INFO(NCCL_INIT|NCCL_NET, "No network plugin found.");
} else {
INFO(NCCL_INIT, "Unable to load libnccl-net.so : %s", dlerror());
INFO(NCCL_INIT|NCCL_NET, "Unable to load libnccl-net.so : %s", dlerror());
}
return ncclSuccess;
}
ncclNet_t* extNet = (ncclNet_t*) dlsym(netPluginLib, STR(NCCL_PLUGIN_SYMBOL));
if (extNet == NULL) {
INFO(NCCL_INIT, "NetPlugin: could not find " STR(NCCL_PLUGIN_SYMBOL) " symbol");
INFO(NCCL_INIT|NCCL_NET, "NetPlugin: could not find " STR(NCCL_PLUGIN_SYMBOL) " symbol");
goto cleanup;
}
if (initNet(extNet) == ncclSuccess) {
@ -116,7 +116,7 @@ ncclResult_t initNet() {
NCCLCHECK(initNetPlugin(&ncclNet));
if (ncclNet != NULL) {
INFO(NCCL_INIT, "Using network plugin %s", ncclNetName());
INFO(NCCL_INIT|NCCL_NET, "Using network plugin %s", ncclNetName());
return ncclSuccess;
}
if (initNet(&ncclNetIb) == ncclSuccess) {
@ -124,7 +124,7 @@ ncclResult_t initNet() {
} else {
ncclNet = &ncclNetSocket;
}
INFO(NCCL_INIT,"Using network %s", ncclNetName());
INFO(NCCL_INIT|NCCL_NET,"Using network %s", ncclNetName());
return ncclSuccess;
}