NPKit: doubling size of event buffers following MAXCHANNELS change (#1135)

Цей коміт міститься в:
Wenkai Du
2024-04-09 08:02:58 -07:00
зафіксовано GitHub
джерело 137571fa01
коміт 0ce68f21d4
+2 -2
Переглянути файл
@@ -20,9 +20,9 @@
class NpKit {
public:
static const uint64_t kNumGpuEventBuffers = 512;
static const uint64_t kNumGpuEventBuffers = 1024;
static const uint64_t kNumCpuEventBuffers = 32;
static const uint64_t kNumCpuEventBuffers = 64;
static ncclResult_t Init(int rank);