From 0ce68f21d4310f283d1f13f65520b8f57481082a Mon Sep 17 00:00:00 2001
From: Wenkai Du <43822138+wenkaidu@users.noreply.github.com>
Date: Tue, 9 Apr 2024 08:02:58 -0700
Subject: [PATCH] NPKit: doubling size of event buffers following MAXCHANNELS
 change (#1135)

---
 src/include/npkit/npkit.h | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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);