diff --git a/src/include/npkit/npkit.h b/src/include/npkit/npkit.h index c535b2ccde..30c3a77ac6 100644 --- a/src/include/npkit/npkit.h +++ b/src/include/npkit/npkit.h @@ -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);