From 6a43b64d0cf8bdd637aa83a1d92ff68f4e4cee4b Mon Sep 17 00:00:00 2001
From: "Poag, Charis" <Charis.Poag@amd.com>
Date: Mon, 28 Apr 2025 14:16:07 -0500
Subject: [PATCH] [SWDEV-518561] P1: Inconsistent memory partition change
 (#315)

Changes:
- Cleaned up drm connections before changing memory partitions

Change-Id: I08430d22536bb6094c91a06fbf2f74d9b75a4fa7

Signed-off-by: Charis Poag <Charis.Poag@amd.com>
---
 src/amd_smi/amd_smi.cc | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/amd_smi/amd_smi.cc b/src/amd_smi/amd_smi.cc
index c015364692..8df8c10ec7 100644
--- a/src/amd_smi/amd_smi.cc
+++ b/src/amd_smi/amd_smi.cc
@@ -1930,6 +1930,8 @@ amdsmi_set_gpu_memory_partition(amdsmi_processor_handle processor_handle,
     } else if (it == nps_amdsmi_to_RSMI.end()) {
         return AMDSMI_STATUS_INVAL;
     }
+
+    amd::smi::AMDSmiSystem::getInstance().clean_up_drm();
     amdsmi_status_t ret = rsmi_wrapper(rsmi_dev_memory_partition_set, processor_handle, 0,
                                         rsmi_type);