From 5f2992a546060db5d83c07571508f0bf499ebc12 Mon Sep 17 00:00:00 2001
From: Tao Sang <tao.sang@amd.com>
Date: Thu, 2 Jan 2025 19:04:52 -0500
Subject: [PATCH] SWDEV-506090 - Fix Unit_hipMemAdvise_TstAlignedAllocMem

Change-Id: I651c25f011d24d0190c62586722b81ddc580655f
---
 catch/unit/memory/hipMemAdvise_old.cc | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/catch/unit/memory/hipMemAdvise_old.cc b/catch/unit/memory/hipMemAdvise_old.cc
index fcf98258fb..9b785aa61a 100644
--- a/catch/unit/memory/hipMemAdvise_old.cc
+++ b/catch/unit/memory/hipMemAdvise_old.cc
@@ -687,8 +687,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_ERROR(hipMemPrefetchAsync(Mllc, MemSz, 0, strm),
-                      hipErrorInvalidValue);
+      HIP_CHECK(hipMemPrefetchAsync(Mllc, MemSz, 0, strm));
       HIP_CHECK(hipStreamSynchronize(strm));
       MemAdvise2<<<(NumElms/32), 32, 0, strm>>>(Mllc, NumElms);
       HIP_CHECK(hipStreamSynchronize(strm));