Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/marian-nmt/nccl.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
Diffstat (limited to 'src/transport/net_ib.cc')
-rw-r--r--src/transport/net_ib.cc2
1 files changed, 2 insertions, 0 deletions
diff --git a/src/transport/net_ib.cc b/src/transport/net_ib.cc
index 1a832f2..97eca9f 100644
--- a/src/transport/net_ib.cc
+++ b/src/transport/net_ib.cc
@@ -112,6 +112,7 @@ static int ncclIbSpeed(int speed) {
}
ncclResult_t ncclIbInit(ncclDebugLogger_t logFunction) {
+ static int shownIbHcaEnv = 0;
if(wrap_ibv_symbols() != ncclSuccess) { return ncclInternalError; }
if (ncclParamIbDisable()) return ncclInternalError;
@@ -131,6 +132,7 @@ ncclResult_t ncclIbInit(ncclDebugLogger_t logFunction) {
// Check if user defined which IB device:port to use
char* userIbEnv = getenv("NCCL_IB_HCA");
+ if (userIbEnv != NULL && shownIbHcaEnv++ == 0) INFO(NCCL_NET|NCCL_ENV, "NCCL_IB_HCA set to %s", userIbEnv);
struct netIf userIfs[MAX_IB_DEVS];
bool searchNot = userIbEnv && userIbEnv[0] == '^';
if (searchNot) userIbEnv++;