Print NCCL version at start
This commit is contained in:
@@ -915,6 +915,7 @@ int main(int argc, char* argv[]) {
|
||||
NCCLCHECK(ncclGroupEnd());
|
||||
}
|
||||
|
||||
PRINT("# NCCL Tests compiled with NCCL %d.%d\n", NCCL_MAJOR, NCCL_MINOR);
|
||||
PRINT("# Using devices\n");
|
||||
for (int p=0; p<nProcs; p++) {
|
||||
if (p == proc) {
|
||||
|
||||
@@ -8,6 +8,9 @@
|
||||
#define NCCL1_COMPAT_H
|
||||
|
||||
#ifndef NCCL_MAJOR // NCCL 1.x
|
||||
#define NCCL_MAJOR 1
|
||||
#define NCCL_MINOR 0
|
||||
|
||||
#define ncclNumOps nccl_NUM_OPS
|
||||
#define ncclNumTypes nccl_NUM_TYPES
|
||||
|
||||
|
||||
Fai riferimento in un nuovo problema
Block a user