From 91fef1215cc89b9c3adb1db5401ea5ad35b05323 Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Wed, 1 Mar 2023 10:10:08 +0000
Subject: [PATCH] SWDEV-362365 - Return invalid value if config is not valid.

Change-Id: I5f7300060f25536f8d3fd25afc2fe2522b621432


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

diff --git a/projects/clr/hipamd/src/hip_device_runtime.cpp b/projects/clr/hipamd/src/hip_device_runtime.cpp
index 5376ae0faf..3fe2703e3f 100644
--- a/projects/clr/hipamd/src/hip_device_runtime.cpp
+++ b/projects/clr/hipamd/src/hip_device_runtime.cpp
@@ -503,7 +503,11 @@ hipError_t hipDeviceSetLimit ( hipLimit_t limit, size_t value ) {
 
 hipError_t hipDeviceSetSharedMemConfig ( hipSharedMemConfig config ) {
   HIP_INIT_API(hipDeviceSetSharedMemConfig, config);
-
+  if (config != hipSharedMemBankSizeDefault &&
+      config != hipSharedMemBankSizeFourByte && 
+      config != hipSharedMemBankSizeEightByte) {
+    HIP_RETURN(hipErrorInvalidValue);
+  }
   // No way to set cache config yet.
 
   HIP_RETURN(hipErrorNotSupported);