From b21a5c64f6e7f90ff4d29a017bdc31e92a413e2d Mon Sep 17 00:00:00 2001
From: Sarbojit Sarkar <Sarbojit.Sarkar@amd.com>
Date: Thu, 10 Jun 2021 06:41:23 +0000
Subject: [PATCH] SWDEV-286147 - Added image support check before using

Change-Id: I4b8ae854bf3488ef2418f0864a15b2fef0a889da


[ROCm/clr commit: 4c5c0419be0fdd1851798dbd3d6e1403c03638ca]
---
 projects/clr/hipamd/src/hip_memory.cpp | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/projects/clr/hipamd/src/hip_memory.cpp b/projects/clr/hipamd/src/hip_memory.cpp
index 0d49cb942c..88aca2a919 100755
--- a/projects/clr/hipamd/src/hip_memory.cpp
+++ b/projects/clr/hipamd/src/hip_memory.cpp
@@ -630,6 +630,11 @@ hipError_t ihipMallocPitch(void** ptr, size_t* pitch, size_t width, size_t heigh
     return hipSuccess;
   }
 
+  if (device && !device->info().imageSupport_) {
+    LogPrintfError("Image is not supported on device %p \n", device);
+    return hipErrorInvalidValue;
+  }
+
   const amd::Image::Format imageFormat(*image_format);
 
   *pitch = amd::alignUp(width * imageFormat.getElementSize(), device->info().imagePitchAlignment_);