diff --git a/vdi/hip_memory.cpp b/vdi/hip_memory.cpp
index 2aa54aca30..5d568e26a4 100644
--- a/vdi/hip_memory.cpp
+++ b/vdi/hip_memory.cpp
@@ -204,7 +204,7 @@ hipError_t hipHostMalloc(void** ptr, size_t sizeBytes, unsigned int flags) {
   }
 
   unsigned int ihipFlags = CL_MEM_SVM_FINE_GRAIN_BUFFER | (flags << 16);
-  if (flags & hipHostMallocCoherent ||
+  if (flags & (hipHostMallocCoherent | hipHostMallocMapped) ||
      (!(flags & hipHostMallocNonCoherent) && HIP_HOST_COHERENT)) {
     ihipFlags |= CL_MEM_SVM_ATOMICS;
   }