From b8a9a32ccb889d00ffd060601f6c825a9fb05a2f Mon Sep 17 00:00:00 2001
From: Sylvain Jeaugey <sjeaugey@nvidia.com>
Date: Tue, 4 Dec 2018 13:10:19 -0800
Subject: [PATCH] Add NCCL_NET flag to many debug lines.

---
 src/init.cu | 12 ++++++------
 1 file 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;
 }