Fix some memory leaks

Bu işleme şunda yer alıyor:
Luke Yeager
2020-01-07 13:30:19 -08:00
işlemeyi yapan: Sylvain Jeaugey
ebeveyn 7a833631b2
işleme af4fa0f4cf
+3 -1
Dosyayı Görüntüle
@@ -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;
}