SWDEV-374373 - Occupied size should be within range.

Change-Id: I20e24c628c51154c2f8f36dca0a6fa4ead1ee6d0
2023-01-02 11:12:52 +00:00
parent 067c7316ba
commit 67792dd858
1 changed files with 8 additions and 0 deletions
@@ -2760,6 +2760,14 @@ hipError_t ihipGraphMemsetParams_validate(const hipMemsetParams* pNodeParams) {
  if (pNodeParams->height <= 0) {
    return hipErrorInvalidValue;
  }
+
+  amd::Memory *memObj = amd::MemObjMap::FindMemObj(pNodeParams->dst);
+  if (memObj != nullptr) {
+    if ((pNodeParams->pitch * pNodeParams->height) > memObj->getSize()) {
+      return hipErrorInvalidValue;
+    }
+  }
+
  return hipSuccess;
 }