kanshan/nccl-tests
1
0
Fork 0
mirror of https://github.com/NVIDIA/nccl-tests.git synced 2026-04-25 08:58:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix some memory leaks

Browse Source
This commit is contained in:
Luke Yeager 2020-01-07 13:30:19 -08:00 committed by Sylvain Jeaugey
parent 7a833631b2
commit af4fa0f4cf
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/common.cu
View File

@ -302,7 +302,8 @@ testResult_t CheckData(struct threadArgs* args, ncclDataType_t type, ncclRedOp_t
printf("%d:%d ", j, dataHost[j]);
}
printf("\n");
free(temp);
free(expectedHost);
free(dataHost);
}
#endif
}
@ -351,6 +352,7 @@ testResult_t testStreamSynchronize(int ngpus, cudaStream_t* streams, ncclComm_t*
// We might want to let other threads (including NCCL threads) use the CPU.
if (idle) pthread_yield();
}
free(done);
return testSuccess;
}