From 3e8f6758e5e12e69d715ff7d9da8a360bb9a1eff Mon Sep 17 00:00:00 2001
From: Saeed Maleki <saemal@microsoft.com>
Date: Sun, 19 Mar 2023 06:35:40 +0000
Subject: [PATCH] both allgather algorithms

---
 tests/allgather_test.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/allgather_test.cu b/tests/allgather_test.cu
index d26e30f6..1bb0635a 100644
--- a/tests/allgather_test.cu
+++ b/tests/allgather_test.cu
@@ -180,7 +180,7 @@ int main(int argc, const char *argv[])
 
   int *data_d;
   uint64_t *flag_d;
-  size_t data_size = 1024*1024*1024;
+  size_t data_size = 1024*1024;
   int nelemsPerGPU = data_size / sizeof(int) / world_size;
   CUDACHECK(cudaMalloc(&data_d, data_size));
   CUDACHECK(cudaMalloc(&flag_d, sizeof(uint64_t)));