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_);