From 6411ca1f6d0e24d41ef182614da71e378bd14dff Mon Sep 17 00:00:00 2001
From: Lakhan Singh <lakhan-singh.thakur@amd.com>
Date: Thu, 3 May 2018 09:27:50 +0530
Subject: [PATCH] Null checks added for hipmallocpitch and hipmemcpy apis

---
 src/hip_memory.cpp | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/hip_memory.cpp b/src/hip_memory.cpp
index 3219dac3db..48a8da81ca 100644
--- a/src/hip_memory.cpp
+++ b/src/hip_memory.cpp
@@ -339,6 +339,11 @@ hipError_t hipHostAlloc(void** ptr, size_t sizeBytes, unsigned int flags) {
 // width in bytes
 hipError_t ihipMallocPitch(void** ptr, size_t* pitch, size_t width, size_t height, size_t depth) {
     hipError_t hip_status = hipSuccess;
+    if(ptr==NULL)
+     {
+	hip_status=hipErrorInvalidValue;
+       	return hip_status;
+     }
     // hardcoded 128 bytes
     *pitch = ((((int)width - 1) / 128) + 1) * 128;
     const size_t sizeBytes = (*pitch) * height;
@@ -1132,7 +1137,11 @@ hipError_t hipMemcpy(void* dst, const void* src, size_t sizeBytes, hipMemcpyKind
     hc::completion_future marker;
 
     hipError_t e = hipSuccess;
-
+    if(dst==NULL || src==NULL)
+	{
+	e=hipErrorInvalidValue;
+	return e;
+	}
     try {
         stream->locked_copySync(dst, src, sizeBytes, kind);
     } catch (ihipException& ex) {