From 409f23d905c66befc78fd3b136595736a5ad3adf Mon Sep 17 00:00:00 2001
From: Satyanvesh Dittakavi <Satyanvesh.Dittakavi@amd.com>
Date: Wed, 21 Dec 2022 11:48:52 +0000
Subject: [PATCH] SWDEV-374371 - hipGraphAddMemsetNode should return error if
 width is 0

Change-Id: I1cd4be271f65c9cade8dddb6d00191e1b74c96bc
---
 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 219b5fedab..b3c8865380 100644
--- a/hipamd/src/hip_memory.cpp
+++ b/hipamd/src/hip_memory.cpp
@@ -2748,6 +2748,11 @@ hipError_t ihipGraphMemsetParams_validate(const hipMemsetParams* pNodeParams) {
   if (pNodeParams == nullptr) {
     return hipErrorInvalidValue;
   }
+
+  if (pNodeParams->width == 0) {
+    return hipErrorInvalidValue;
+  }
+
   if (pNodeParams->elementSize != 1 && pNodeParams->elementSize != 2 && pNodeParams->elementSize != 4) {
     return hipErrorInvalidValue;
   }