microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-13 01:36:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

both allgather algorithms

Browse Source
This commit is contained in:
Saeed Maleki
2023-03-19 06:35:40 +00:00
parent 17cbc84a14
commit 3e8f6758e5
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/allgather_test.cu
View File

@@ -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)));