Sylvain Jeaugey
353e8ba446
Fix bug with CollNet Fix bug with zero-bytes send/recv operations Fix NCCL_PARAM implementation to avoid taking a lock on every call Fix bug when setting NCCL_IB_QPS_PER_CONNECTION to more than one. Improve error reporting for network errors.
30 lines
981 B
C++
30 lines
981 B
C++
/*************************************************************************
|
|
* Copyright (c) 2017-2022, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#ifndef NCCL_PARAM_H_
|
|
#define NCCL_PARAM_H_
|
|
|
|
#include <stdint.h>
|
|
|
|
const char* userHomeDir();
|
|
void setEnvFile(const char* fileName);
|
|
void initEnv();
|
|
|
|
void ncclLoadParam(char const* env, int64_t deftVal, int64_t uninitialized, int64_t* cache);
|
|
|
|
#define NCCL_PARAM(name, env, deftVal) \
|
|
int64_t ncclParam##name() { \
|
|
constexpr int64_t uninitialized = INT64_MIN; \
|
|
static_assert(deftVal != uninitialized, "default value cannot be the uninitialized value."); \
|
|
static int64_t cache = uninitialized; \
|
|
if (__builtin_expect(__atomic_load_n(&cache, __ATOMIC_RELAXED) == uninitialized, false)) { \
|
|
ncclLoadParam("NCCL_" env, deftVal, uninitialized, &cache); \
|
|
} \
|
|
return cache; \
|
|
}
|
|
|
|
#endif
|