NASP/cuda-samples
34
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2024-10-06 10:29:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

add cudaDeviceReset() into p2pBandwidthLatencyTest to free gpu memory after test

Browse Source
This commit is contained in:
Konnase Lee 2019-08-10 16:42:49 +08:00
parent 337815dbee
commit f53f159886
1 changed files with 6 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
Samples/p2pBandwidthLatencyTest/p2pBandwidthLatencyTest.cu
View File

@ -277,10 +277,9 @@ void outputBandwidthMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) {
cudaCheckError();
cudaStreamDestroy(stream[d]);
cudaCheckError();
cudaDeviceReset();
cudaCheckError();
}
cudaFreeHost((void *)flag);
cudaCheckError();
}
void outputBidirectionalBandwidthMatrix(int numGPUs, bool p2p) {
@ -433,10 +432,9 @@ void outputBidirectionalBandwidthMatrix(int numGPUs, bool p2p) {
cudaCheckError();
cudaStreamDestroy(stream1[d]);
cudaCheckError();
cudaDeviceReset();
cudaCheckError();
}
cudaFreeHost((void *)flag);
cudaCheckError();
}
void outputLatencyMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) {
@ -586,12 +584,11 @@ void outputLatencyMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) {
cudaCheckError();
cudaStreamDestroy(stream[d]);
cudaCheckError();
cudaDeviceReset();
cudaCheckError();
}
sdkDeleteTimer(&stopWatch);
cudaFreeHost((void *)flag);
cudaCheckError();
}
int main(int argc, char **argv) {