NPKit: doubling size of event buffers following MAXCHANNELS change (#1135)
This commit is contained in:
@@ -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);
|
||||
|
||||
|
||||
Reference in New Issue
Block a user