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

This commit is contained in:
Wenkai Du
2024-04-09 08:02:58 -07:00
committed by GitHub
szülő 137571fa01
commit 0ce68f21d4
+2 -2
Fájl megtekintése
@@ -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);