diff options
| author | Sylvain Jeaugey <sjeaugey@nvidia.com> | 2018-12-05 00:10:19 +0300 |
|---|---|---|
| committer | Sylvain Jeaugey <sjeaugey@nvidia.com> | 2018-12-05 00:10:19 +0300 |
| commit | b8a9a32ccb889d00ffd060601f6c825a9fb05a2f (patch) | |
| tree | 07df9d1fd1fe2a70c12c460c1f65f8e68e978648 | |
| parent | cdae05b27728b1b5b25b3a4f294334df9fb5164d (diff) | |
Add NCCL_NET flag to many debug lines.
| -rw-r--r-- | src/init.cu | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/src/init.cu b/src/init.cu index a8c8011..5663323 100644 --- a/src/init.cu +++ b/src/init.cu @@ -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; } |