From 3cef4c8a97aadb6a4716260f2868dc843f722a53 Mon Sep 17 00:00:00 2001
From: agodavar <Anusha.GodavarthySurya@amd.com>
Date: Mon, 2 Nov 2020 11:45:03 -0500
Subject: [PATCH] SWDEV-256810 : hipMemcpy2D API handle width, src/dst pitch 0

Change-Id: I4f60b220624f4235c0c3daf68adcd5a496e93696


[ROCm/hip commit: 77e5c4110e78d18aa1c0555946aae25525db425c]
---
 projects/hip/rocclr/hip_memory.cpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/projects/hip/rocclr/hip_memory.cpp b/projects/hip/rocclr/hip_memory.cpp
index 00854dbde4..f6dc7ce070 100755
--- a/projects/hip/rocclr/hip_memory.cpp
+++ b/projects/hip/rocclr/hip_memory.cpp
@@ -1511,6 +1511,12 @@ hipError_t ihipMemcpyParam2D(const hip_Memcpy2D* pCopy,
 hipError_t ihipMemcpy2D(void* dst, size_t dpitch, const void* src, size_t spitch, size_t width,
                         size_t height, hipMemcpyKind kind, hipStream_t stream, bool isAsync = false) {
   hip_Memcpy2D desc = {};
+  if (spitch == 0 || dpitch == 0) {
+    return hipErrorUnknown;
+  }
+  if (width == 0) {
+    return hipSuccess;
+  }
 
   desc.srcXInBytes = 0;
   desc.srcY = 0;