From c4003ef62a2b9d5bf1f3e2812b40d58f98434568 Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Wed, 21 Dec 2022 18:07:00 +0000
Subject: [PATCH] SWDEV-374398 - Pitch should not be less than width for both
 src and dst.

Change-Id: Ib8fa8e3bd0c655f3de213b3798d369ba866b2f01
---
 hipamd/src/hip_memory.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/hipamd/src/hip_memory.cpp b/hipamd/src/hip_memory.cpp
index dc8b3464c9..1a6272c7f9 100644
--- a/hipamd/src/hip_memory.cpp
+++ b/hipamd/src/hip_memory.cpp
@@ -2496,6 +2496,11 @@ hipError_t ihipMemcpy3D_validate(const hipMemcpy3DParms* p) {
       (hip::getElementSize(p->dstArray) != hip::getElementSize(p->dstArray))) {
     return hipErrorInvalidValue;
   }
+
+  // Pitch should not be less than width for both src and dst.
+  if (p->srcPtr.pitch < p->srcPtr.xsize || p->dstPtr.pitch < p->dstPtr.xsize) {
+    return hipErrorInvalidPitchValue;
+  }
   return hipSuccess;
 }