From e72d5a4443c1fdb998b8d882efad430f2d2aec0d Mon Sep 17 00:00:00 2001
From: kjayapra-amd <karthik.jayaprakash@amd.com>
Date: Mon, 5 Aug 2024 11:11:54 -0400
Subject: [PATCH] SWDEV-439234 - set access for vmm memory on graph/mempool
 path.

Change-Id: Idfb740dcfe6c7fe0f18231de3074a81d06e6886e
---
 hipamd/src/hip_graph_internal.hpp | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/hipamd/src/hip_graph_internal.hpp b/hipamd/src/hip_graph_internal.hpp
index 4e0cce378c..ba7bc8d84f 100644
--- a/hipamd/src/hip_graph_internal.hpp
+++ b/hipamd/src/hip_graph_internal.hpp
@@ -639,6 +639,12 @@ struct Graph {
       LogError("Failed to reserve Virtual Address");
     }
 
+    // Set Access to read write for all devices.
+    for (size_t dev_idx = 0; dev_idx < g_devices.size(); ++dev_idx) {
+      amd::Device* device = g_devices[dev_idx]->devices()[0];
+      device->SetMemAccess(ptr, size, amd::Device::VmmAccess::kReadWrite);
+    }
+
     return ptr;
   }