From ee80c2cac4a44c26966b94686752fd4ed444c2cf Mon Sep 17 00:00:00 2001
From: Maisam Arif <maisarif@amd.com>
Date: Wed, 24 Jan 2024 02:05:01 -0600
Subject: [PATCH] Handled unkown vram type out of bounds error

Signed-off-by: Maisam Arif <maisarif@amd.com>
Change-Id: I2d32c7043c78c0651f1b4db565a299b6b96abbcc
---
 src/amd_smi/amd_smi.cc | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/amd_smi/amd_smi.cc b/src/amd_smi/amd_smi.cc
index caa1f2884c..4d3e696d23 100644
--- a/src/amd_smi/amd_smi.cc
+++ b/src/amd_smi/amd_smi.cc
@@ -742,6 +742,10 @@ amdsmi_status_t amdsmi_get_gpu_vram_info(
         }
     }
 
+    // if vram type is greater than the max enum set it to unknown
+    if (info->vram_type > VRAM_TYPE__MAX)
+        info->vram_type = VRAM_TYPE_UNKNOWN;
+
     // map the vendor name to enum
     char brand[256];
     r = rsmi_wrapper(rsmi_dev_vram_vendor_get, processor_handle, brand, 255);