Merge f53f1598861be4794fdb7e6cc42ce36d1eb2cf93 into 11de19f00cd24e244d2f6869c64810d63aafb926

This commit is contained in:
Konnase Lee 2021-12-03 17:59:07 +08:00 committed by GitHub
commit 37db718c06
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -281,10 +281,9 @@ void outputBandwidthMatrix(int numElems, int numGPUs, bool p2p, P2PDataTransfer
cudaCheckError(); cudaCheckError();
cudaStreamDestroy(stream[d]); cudaStreamDestroy(stream[d]);
cudaCheckError(); cudaCheckError();
cudaDeviceReset();
cudaCheckError();
} }
cudaFreeHost((void *)flag);
cudaCheckError();
} }
void outputBidirectionalBandwidthMatrix(int numElems, int numGPUs, bool p2p) { void outputBidirectionalBandwidthMatrix(int numElems, int numGPUs, bool p2p) {
@ -438,10 +437,9 @@ void outputBidirectionalBandwidthMatrix(int numElems, int numGPUs, bool p2p) {
cudaCheckError(); cudaCheckError();
cudaStreamDestroy(stream1[d]); cudaStreamDestroy(stream1[d]);
cudaCheckError(); cudaCheckError();
cudaDeviceReset();
cudaCheckError();
} }
cudaFreeHost((void *)flag);
cudaCheckError();
} }
void outputLatencyMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) { void outputLatencyMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) {
@ -594,12 +592,11 @@ void outputLatencyMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) {
cudaCheckError(); cudaCheckError();
cudaStreamDestroy(stream[d]); cudaStreamDestroy(stream[d]);
cudaCheckError(); cudaCheckError();
cudaDeviceReset();
cudaCheckError();
} }
sdkDeleteTimer(&stopWatch); sdkDeleteTimer(&stopWatch);
cudaFreeHost((void *)flag);
cudaCheckError();
} }
int main(int argc, char **argv) { int main(int argc, char **argv) {