add cudaDeviceReset() into p2pBandwidthLatencyTest to free gpu memory…

… after test
NVIDIA · Aug 10, 2019 · 6661519 · 6661519
1 parent 337815d
commit 6661519
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/Samples/p2pBandwidthLatencyTest/p2pBandwidthLatencyTest.cu b/Samples/p2pBandwidthLatencyTest/p2pBandwidthLatencyTest.cu
@@ -143,6 +143,7 @@ void performP2PCopy(int *dest, int destDevice, int *src, int srcDevice,
                           sizeof(int) * num_elems, streamToRun);
     }
   }
+  cudaDeviceReset();
 }
 
 void outputBandwidthMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) {
@@ -280,6 +281,7 @@ void outputBandwidthMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) {
   }
 
   cudaFreeHost((void *)flag);
+  cudaDeviceReset();
   cudaCheckError();
 }
 
@@ -436,6 +438,7 @@ void outputBidirectionalBandwidthMatrix(int numGPUs, bool p2p) {
   }
 
   cudaFreeHost((void *)flag);
+  cudaDeviceReset();
   cudaCheckError();
 }
 
@@ -591,6 +594,7 @@ void outputLatencyMatrix(int numGPUs, bool p2p, P2PDataTransfer p2p_method) {
   sdkDeleteTimer(&stopWatch);
 
   cudaFreeHost((void *)flag);
+  cudaDeviceReset();
   cudaCheckError();
 }