Sylvain Jeaugey
3c223c105a
Add network communication through another GPU connected with NVLink (PXN). Add aggregation of messages coming from different local GPUs through PXN and going to the same destination. Add new v5 plugin API with grouped receives and tags. Add compat for v4 plugins. Add naming of NCCL threads to help debugging. Fix NVLink detection and avoid data corruption when some NVLinks are down. Add support for Relaxed Ordering for IB. Add profiling and timing infrastructure.
61 строка
1.6 KiB
C
61 строка
1.6 KiB
C
/*************************************************************************
|
|
* Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#ifndef NCCL_TIMER_H_
|
|
#define NCCL_TIMER_H_
|
|
#if ENABLE_TIMER
|
|
#include <unistd.h>
|
|
#include <sys/time.h>
|
|
#include <x86intrin.h>
|
|
static double freq = -1;
|
|
static void calibrate() {
|
|
struct timeval tv;
|
|
gettimeofday(&tv, NULL);
|
|
uint64_t timeCycles = __rdtsc();
|
|
double time = - tv.tv_sec*1E6 - tv.tv_usec;
|
|
uint64_t total = 0ULL;
|
|
for (int i=0; i<10000; i++) total += __rdtsc();
|
|
gettimeofday(&tv, NULL);
|
|
timeCycles = __rdtsc() - timeCycles;
|
|
time += tv.tv_sec*1E6 + tv.tv_usec;
|
|
freq = timeCycles/time;
|
|
}
|
|
static inline double gettime() {
|
|
if (freq == -1) calibrate();
|
|
return __rdtsc()/freq;
|
|
}
|
|
static uint64_t counts[8];
|
|
static double times[8];
|
|
static double startTimes[8];
|
|
#define TIME_START(index) do { \
|
|
counts[index]++; \
|
|
startTimes[index] = gettime(); \
|
|
} while (0);
|
|
|
|
#define TIME_STOP(index) do { \
|
|
times[index] += gettime() - startTimes[index]; \
|
|
} while (0);
|
|
|
|
#define TIME_CANCEL(index) do { \
|
|
counts[index]--; \
|
|
} while (0);
|
|
|
|
#define TIME_PRINT(name) do { \
|
|
printf("%s stats", name); \
|
|
for (int i=0; i<8; i++) { \
|
|
if (counts[i]) printf(" [%d] %g/%ld = %g", i, times[i], counts[i], times[i]/counts[i]); \
|
|
counts[i] = 0; \
|
|
} \
|
|
printf("\n"); \
|
|
} while (0);
|
|
#else
|
|
#define TIME_START(index) while(0);
|
|
#define TIME_STOP(index) while(0);
|
|
#define TIME_CANCEL(index) while(0);
|
|
#define TIME_PRINT(name)
|
|
#endif
|
|
#endif
|