From 545b6e60eef167d0d07daf9d1aa2b8b1504e20ab Mon Sep 17 00:00:00 2001
From: Rahul Garg <rahul.garg@amd.com>
Date: Fri, 7 Jun 2019 05:23:11 +0530
Subject: [PATCH] Add cleanup code in module api samples

[ROCm/hip commit: f2b3526503fbd7bd81105e838c4efc3418479855]
---
 projects/hip/samples/0_Intro/module_api/launchKernelHcc.cpp  | 4 ++++
 projects/hip/samples/0_Intro/module_api/runKernel.cpp        | 4 ++++
 projects/hip/samples/0_Intro/module_api_global/runKernel.cpp | 4 ++++
 3 files changed, 12 insertions(+)

diff --git a/projects/hip/samples/0_Intro/module_api/launchKernelHcc.cpp b/projects/hip/samples/0_Intro/module_api/launchKernelHcc.cpp
index 5ccd0801d1..4ff000d1af 100644
--- a/projects/hip/samples/0_Intro/module_api/launchKernelHcc.cpp
+++ b/projects/hip/samples/0_Intro/module_api/launchKernelHcc.cpp
@@ -105,6 +105,10 @@ int main() {
         std::cout << "FAILED!\n";
     };
 
+    hipFree(Ad);
+    hipFree(Bd);
+    delete A;
+    delete B;
     hipCtxDestroy(context);
     return 0;
 }
diff --git a/projects/hip/samples/0_Intro/module_api/runKernel.cpp b/projects/hip/samples/0_Intro/module_api/runKernel.cpp
index eccc105319..a011b42666 100644
--- a/projects/hip/samples/0_Intro/module_api/runKernel.cpp
+++ b/projects/hip/samples/0_Intro/module_api/runKernel.cpp
@@ -97,6 +97,10 @@ int main() {
         std::cout << "FAILED!\n";
     };
 
+    hipFree(Ad);
+    hipFree(Bd);
+    delete A;
+    delete B;
     hipCtxDestroy(context);
     return 0;
 }
diff --git a/projects/hip/samples/0_Intro/module_api_global/runKernel.cpp b/projects/hip/samples/0_Intro/module_api_global/runKernel.cpp
index 76af21b0f9..1e10dba5c6 100644
--- a/projects/hip/samples/0_Intro/module_api_global/runKernel.cpp
+++ b/projects/hip/samples/0_Intro/module_api_global/runKernel.cpp
@@ -145,6 +145,10 @@ int main() {
         };
     }
 
+    hipFree(Ad);
+    hipFree(Bd);
+    delete A;
+    delete B;
     hipCtxDestroy(context);
     return 0;
 }