From 3eed9aba5d5ad65bc1ae92041b4e0dda146a0a89 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
---
 samples/0_Intro/module_api/runKernel.cpp | 9 ++++++---
 src/hip_module.cpp                       | 1 +
 2 files changed, 7 insertions(+), 3 deletions(-)

diff --git a/samples/0_Intro/module_api/runKernel.cpp b/samples/0_Intro/module_api/runKernel.cpp
index e7d54beb54..355f0bf5da 100644
--- a/samples/0_Intro/module_api/runKernel.cpp
+++ b/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);
 
diff --git a/src/hip_module.cpp b/src/hip_module.cpp
index 554e13387a..67bba5f935 100644
--- a/src/hip_module.cpp
+++ b/src/hip_module.cpp
@@ -403,6 +403,7 @@ hipError_t ihipModuleLaunchKernel(hipFunction_t f,
             } else {
                 return ihipLogStatus(hipErrorNotInitialized);
             }
+            
         }else{
             return ihipLogStatus(hipErrorInvalidValue);
         }