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

fix the size of the recv buffer in AllGather UBR test (#1564)

Browse Source 
[ROCm/rccl commit: 59c55842f1]
This commit is contained in:
isaki001
2025-03-05 11:42:15 -06:00
committed by GitHub
parent e7c90e0a46
commit 14be1c9a7a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
projects/rccl/test/AllGatherTests.cpp
+2 -2
View File
@@ -127,7 +127,7 @@ namespace RcclUnitTesting
const int nranks = 8;
size_t count = 2048;
std::vector<int> sendBuff(count, 0);
std::vector<int> recvBuff(count, 0);
std::vector<int> recvBuff(nranks*count, 0);
std::vector<int> expected(nranks*count, 0);
for (int i = 0; i < count; ++i){
@@ -146,7 +146,7 @@ namespace RcclUnitTesting
const int nranks = 8;
size_t count = 2048;
std::vector<int> sendBuff(count, 0);
std::vector<int> recvBuff(count, 0);
std::vector<int> recvBuff(nranks*count, 0);
std::vector<int> expected(nranks*count, 0);
const bool use_managed_mem = true;
for (int i = 0; i < count; ++i){