From 59c55842f1e934e2857c720098b395eb12aec4f7 Mon Sep 17 00:00:00 2001 From: isaki001 <36317038+isaki001@users.noreply.github.com> Date: Wed, 5 Mar 2025 11:42:15 -0600 Subject: [PATCH] fix the size of the recv buffer in AllGather UBR test (#1564) --- test/AllGatherTests.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/AllGatherTests.cpp b/test/AllGatherTests.cpp index a59a3522d4..a712673af7 100644 --- a/test/AllGatherTests.cpp +++ b/test/AllGatherTests.cpp @@ -127,7 +127,7 @@ namespace RcclUnitTesting const int nranks = 8; size_t count = 2048; std::vector sendBuff(count, 0); - std::vector recvBuff(count, 0); + std::vector recvBuff(nranks*count, 0); std::vector 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 sendBuff(count, 0); - std::vector recvBuff(count, 0); + std::vector recvBuff(nranks*count, 0); std::vector expected(nranks*count, 0); const bool use_managed_mem = true; for (int i = 0; i < count; ++i){