From 7a572e016230a81203ffa86fded9e2d1e36c52ef Mon Sep 17 00:00:00 2001
From: Rakesh Roy <rakesh.roy@amd.com>
Date: Thu, 21 Apr 2022 21:05:12 +0530
Subject: [PATCH] SWDEV-332604 - Add nullptr check for argument pitch

Change-Id: I6d7aa5356a74b92f21f7a51a7d3e2f8d0b920e07


[ROCm/clr commit: 2613ebf157173dad9cdb4d1f23b9696e71744faa]
---
 projects/clr/hipamd/src/hip_memory.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/clr/hipamd/src/hip_memory.cpp b/projects/clr/hipamd/src/hip_memory.cpp
index 1cbbf40f88..ef08b92955 100644
--- a/projects/clr/hipamd/src/hip_memory.cpp
+++ b/projects/clr/hipamd/src/hip_memory.cpp
@@ -653,7 +653,7 @@ hipError_t ihipMallocPitch(void** ptr, size_t* pitch, size_t width, size_t heigh
 
   amd::Device* device = hip::getCurrentDevice()->devices()[0];
 
-  if (ptr == nullptr) {
+  if ((ptr == nullptr) || (pitch == nullptr)) {
     return hipErrorInvalidValue;
   }