SWDEV-352687 - fix AlignedAllocMem in hmm

Change-Id: Icec74ac732a0a6ac2a2600f504aa1ec73717c0ca


[ROCm/clr commit: c9ce9b3300]

projects/clr/hipamd/src/hip_hmm.cpp

@@ -87,10 +87,9 @@ hipError_t hipMemPrefetchAsync(const void* dev_ptr, size_t count, int device,
   size_t offset = 0;
   amd::Memory* memObj = getMemoryObject(dev_ptr, offset);
 
-  if (memObj == nullptr || (count  > (memObj->getSize() - offset))) {
+  if ((memObj != nullptr) && (count  > (memObj->getSize() - offset))) {
     HIP_RETURN(hipErrorInvalidValue);
   }
-
   if (device != hipCpuDeviceId && (static_cast<size_t>(device) >= g_devices.size())) {
     HIP_RETURN(hipErrorInvalidDevice);
   }
@@ -99,6 +98,11 @@ hipError_t hipMemPrefetchAsync(const void* dev_ptr, size_t count, int device,
   amd::Device* dev = nullptr;
   bool cpu_access = false;
 
+  if ((memObj == nullptr) && (device != hipCpuDeviceId) &&
+      (!g_devices[device]->devices()[0]->info().hmmCpuMemoryAccessible_)) {
+    HIP_RETURN(hipErrorNotSupported);
+  }
+
   // Pick the specified stream or Null one from the provided device
   if (device == hipCpuDeviceId) {
     cpu_access = true;