From d1507ad4755dd9f914f92c8d2137495daa4309f0 Mon Sep 17 00:00:00 2001
From: German Andryeyev <German.Andryeyev@amd.com>
Date: Tue, 23 Apr 2024 15:54:28 -0400
Subject: [PATCH] SWDEV-353281 - Align VA size

Lower layer ignores alignment

Change-Id: If16df951ecefddc804a6effe013058afc595d30f


[ROCm/clr commit: 5c2344019942b6e988541444758aec90be9910bb]
---
 projects/clr/hipamd/src/hip_graph_internal.hpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/projects/clr/hipamd/src/hip_graph_internal.hpp b/projects/clr/hipamd/src/hip_graph_internal.hpp
index 9c4cd083a1..21737110f7 100644
--- a/projects/clr/hipamd/src/hip_graph_internal.hpp
+++ b/projects/clr/hipamd/src/hip_graph_internal.hpp
@@ -495,6 +495,7 @@ struct Graph {
     void* ptr;
     const auto& dev_info = g_devices[0]->devices()[0]->info();
 
+    size = amd::alignUp(size, dev_info.virtualMemAllocGranularity_);
     // Single virtual alloc would reserve for all devices.
     ptr = g_devices[0]->devices()[0]->virtualAlloc(startAddress, size,
             dev_info.virtualMemAllocGranularity_);