AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source 
[ROCm/rccl commit: 0ce68f21d4]
This commit is contained in:
Wenkai Du
2024-04-09 08:02:58 -07:00
committed by GitHub
parent 0941d6bc6e
commit 99c7fc29ba
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/src/include/npkit/npkit.h
+2 -2
View File
@@ -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);