Call cudaFreeHost() on wrongPerGpu not cudaFree()

This commit is contained in:
David Addison 2022-11-22 11:18:37 -08:00
parent 3bd2bd292b
commit 24fcf64ed1

View File

@ -262,7 +262,7 @@ testResult_t CheckData(struct threadArgs* args, ncclDataType_t type, ncclRedOp_t
*wrongElts = 0;
for (int i=0; i < args->nGpus; i++) *wrongElts += wrongPerGpu[i];
cudaFree(wrongPerGpu);
cudaFreeHost(wrongPerGpu);
if (args->reportErrors && *wrongElts) args->errors[0]++;
return testSuccess;