From 8f495da889102008ebb7a6fd4ba67d19fe8bf628 Mon Sep 17 00:00:00 2001
From: Aditya Atluri <Aditya.Atluri@amd.com>
Date: Fri, 31 Mar 2017 14:13:46 -0500
Subject: [PATCH] added debug support for HIP sample

Change-Id: Ia7265234082039b68114f7421f4dbcb7149d4d2b


[ROCm/hip-tests commit: 93a0b55616853e4a47e717dc93b5bfef0b42da78]
---
 .../hip-tests/samples/0_Intro/module_api/runKernel.cpp   | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/projects/hip-tests/samples/0_Intro/module_api/runKernel.cpp b/projects/hip-tests/samples/0_Intro/module_api/runKernel.cpp
index e7d54beb54..355f0bf5da 100644
--- a/projects/hip-tests/samples/0_Intro/module_api/runKernel.cpp
+++ b/projects/hip-tests/samples/0_Intro/module_api/runKernel.cpp
@@ -33,6 +33,9 @@ THE SOFTWARE.
 #define fileName "vcpy_kernel.code"
 #define kernel_name "hello_world"
 
+#define HIP_CHECK(status) \
+if(status != hipSuccess) {std::cout<<"Got Status: "<<status<<" at Line: "<<__LINE__<<std::endl;exit(0);}
+
 int main(){
     float *A, *B;
     hipDeviceptr_t Ad, Bd;
@@ -57,8 +60,8 @@ int main(){
     hipMemcpyHtoD(Bd, B, SIZE);
     hipModule_t Module;
     hipFunction_t Function;
-    hipModuleLoad(&Module, fileName);
-    hipModuleGetFunction(&Function, Module, kernel_name);
+    HIP_CHECK(hipModuleLoad(&Module, fileName));
+    HIP_CHECK(hipModuleGetFunction(&Function, Module, kernel_name));
 
 #ifdef __HIP_PLATFORM_HCC__
 		uint32_t len = LEN;
@@ -99,7 +102,7 @@ int main(){
       HIP_LAUNCH_PARAM_END
     };
 
-    hipModuleLaunchKernel(Function, 1, 1, 1, LEN, 1, 1, 0, 0, NULL, (void**)&config);
+    HIP_CHECK(hipModuleLaunchKernel(Function, 1, 1, 1, LEN, 1, 1, 0, 0, NULL, (void**)&config));
 
     hipMemcpyDtoH(B, Bd, SIZE);