From f44b862cd94e59d6c81eb7b4282b8a472197340f Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Wed, 26 Jul 2023 06:17:28 +0000
Subject: [PATCH] SWDEV-413161/SWDEV-415848 - Prefetch is only for device
 memory.

Change-Id: Ie2d9cb907fbf7958dfd1b90a5b01a2722cb4e2a5
---
 catch/unit/memory/hipMemAdvise_old.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/catch/unit/memory/hipMemAdvise_old.cc b/catch/unit/memory/hipMemAdvise_old.cc
index b69e46af7b..625c216ef6 100644
--- a/catch/unit/memory/hipMemAdvise_old.cc
+++ b/catch/unit/memory/hipMemAdvise_old.cc
@@ -685,7 +685,7 @@ TEST_CASE("Unit_hipMemAdvise_TstAlignedAllocMem") {
       // The following hipMemAdvise() call is made to know if advise on
       // aligned_alloc() is causing any issue
       HIP_CHECK(hipMemAdvise(Mllc, MemSz, hipMemAdviseSetPreferredLocation, 0));
-      HIP_CHECK(hipMemPrefetchAsync(Mllc, MemSz, 0, strm));
+      HIP_CHECK_ERROR(hipMemPrefetchAsync(Mllc, MemSz, 0, strm), hipErrorInvalidValue);
       HIP_CHECK(hipStreamSynchronize(strm));
       MemAdvise2<<<(NumElms/32), 32, 0, strm>>>(Mllc, NumElms);
       HIP_CHECK(hipStreamSynchronize(strm));
@@ -695,7 +695,7 @@ TEST_CASE("Unit_hipMemAdvise_TstAlignedAllocMem") {
           }
         }
       REQUIRE(DataMismatch == 0);
-      } 
+      }
   } else {
       HipTest::HIP_SKIP_TEST("GPU is not xnack enabled hence skipping the test...\n");
     }