From 01f4d7eac3d3163baa2f05250727446f7554944c Mon Sep 17 00:00:00 2001
From: Ajay <ajay.gunashekar@amd.com>
Date: Thu, 8 Dec 2022 06:55:23 +0000
Subject: [PATCH] SWDEV-369618 -
 hipOccupancyMaxPotentialBlockSizeVariableSMemWithFlags invalid flag scenario

Change-Id: I07df2640be71a6b0aed160d54e80ffdeb85ec554


[ROCm/clr commit: 3e4ed8d9179d21e8dd354873d646b8055f3d4a51]
---
 projects/clr/hipamd/src/hip_platform.cpp | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/projects/clr/hipamd/src/hip_platform.cpp b/projects/clr/hipamd/src/hip_platform.cpp
index 93bdf4f837..ecf238446f 100644
--- a/projects/clr/hipamd/src/hip_platform.cpp
+++ b/projects/clr/hipamd/src/hip_platform.cpp
@@ -586,7 +586,9 @@ hipError_t hipOccupancyMaxActiveBlocksPerMultiprocessorWithFlags(int* numBlocks,
   if (numBlocks == nullptr) {
     HIP_RETURN(hipErrorInvalidValue);
   }
-
+  if (flags != hipOccupancyDefault && flags != hipOccupancyDisableCachingOverride) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
   hipFunction_t func = nullptr;
   hipError_t hip_error = PlatformState::instance().getStatFunc(&func, f, ihipGetDevice());
   if ((hip_error != hipSuccess) || (func == nullptr)) {