Fix some memory leaks

Tento commit je obsažen v:
Luke Yeager
2020-01-07 13:30:19 -08:00
odevzdal Sylvain Jeaugey
rodič 7a833631b2
revize af4fa0f4cf
+3 -1
Zobrazit soubor
@@ -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;
}