From 5ff521d6b942279329dbdfd5c768677f8d935115 Mon Sep 17 00:00:00 2001
From: Mateusz Hoppe <mateusz.hoppe@intel.com>
Date: Mon, 25 Mar 2019 19:19:28 +0100
Subject: [PATCH] Use GPU address in MediaImageSetArgTest

Change-Id: I22477674c4551a8ec2ee0a46942fa7ec764bad02
Signed-off-by: Mateusz Hoppe <mateusz.hoppe@intel.com>
---
 unit_tests/accelerators/media_image_arg_tests.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/unit_tests/accelerators/media_image_arg_tests.cpp b/unit_tests/accelerators/media_image_arg_tests.cpp
index 1f394354eb..83614255b0 100644
--- a/unit_tests/accelerators/media_image_arg_tests.cpp
+++ b/unit_tests/accelerators/media_image_arg_tests.cpp
@@ -119,8 +119,8 @@ HWTEST_F(MediaImageSetArgTest, clSetKernelArgImage) {
         ptrOffset(pKernel->getSurfaceStateHeap(),
                   pKernelInfo->kernelArgInfo[0].offsetHeap));
 
-    void *surfaceAddress = reinterpret_cast<void *>(pSurfaceState->getSurfaceBaseAddress());
-    ASSERT_EQ(srcImage->getCpuAddress(), surfaceAddress);
+    uint64_t surfaceAddress = pSurfaceState->getSurfaceBaseAddress();
+    ASSERT_EQ(srcImage->getGraphicsAllocation()->getGpuAddress(), surfaceAddress);
     EXPECT_EQ(srcImage->getImageDesc().image_width, pSurfaceState->getWidth());
     EXPECT_EQ(srcImage->getImageDesc().image_height, pSurfaceState->getHeight());