From db075a1faba95928eea6ff79e28e413ce8698d84 Mon Sep 17 00:00:00 2001
From: Wenkai Du <wenkai.du@amd.com>
Date: Tue, 16 Nov 2021 07:50:18 -0800
Subject: [PATCH] Add missing hipStreamDestroy at test exit

[ROCm/rccl-tests commit: 602b745ff48f4aa7d73e8fd946442b666fae344f]
---
 projects/rccl-tests/src/common.cu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/projects/rccl-tests/src/common.cu b/projects/rccl-tests/src/common.cu
index 64f5e92263..98fa8dcfb7 100644
--- a/projects/rccl-tests/src/common.cu
+++ b/projects/rccl-tests/src/common.cu
@@ -1268,6 +1268,10 @@ testResult_t run() {
     free(comms);
   }
 
+  for (int i=0; i<nGpus*nThreads; i++) {
+    HIPCHECK(hipStreamDestroy(streams[i]));
+  }
+
   // Free off HIP allocated memory
   for (int i=0; i<nGpus*nThreads; i++) {
     if (memorytype == ncclHost) {