1 files changed, 8 insertions, 4 deletions
diff --git a/src/backends/cl/test/ClMemCopyTests.cpp b/src/backends/cl/test/ClMemCopyTests.cpp
index c26f7bdae8..1048e73c1b 100644
--- a/src/backends/cl/test/ClMemCopyTests.cpp
+++ b/src/backends/cl/test/ClMemCopyTests.cpp
@@ -19,7 +19,8 @@ BOOST_AUTO_TEST_CASE(CopyBetweenCpuAndGpu)
 {
     LayerTestResult<float, 4> result =
         MemCopyTest<armnn::RefWorkloadFactory, armnn::ClWorkloadFactory, armnn::DataType::Float32>(false);
-    auto predResult = CompareTensors(result.output, result.outputExpected);
+    auto predResult = CompareTensors(result.m_ActualData,  result.m_ExpectedData,
+                                     result.m_ActualShape, result.m_ExpectedShape);
     BOOST_TEST(predResult.m_Result, predResult.m_Message.str());
 }
 
@@ -27,7 +28,8 @@ BOOST_AUTO_TEST_CASE(CopyBetweenGpuAndCpu)
 {
     LayerTestResult<float, 4> result =
         MemCopyTest<armnn::ClWorkloadFactory, armnn::RefWorkloadFactory, armnn::DataType::Float32>(false);
-    auto predResult = CompareTensors(result.output, result.outputExpected);
+    auto predResult = CompareTensors(result.m_ActualData,  result.m_ExpectedData,
+                                     result.m_ActualShape, result.m_ExpectedShape);
     BOOST_TEST(predResult.m_Result, predResult.m_Message.str());
 }
 
@@ -35,7 +37,8 @@ BOOST_AUTO_TEST_CASE(CopyBetweenCpuAndGpuWithSubtensors)
 {
     LayerTestResult<float, 4> result =
         MemCopyTest<armnn::RefWorkloadFactory, armnn::ClWorkloadFactory, armnn::DataType::Float32>(true);
-    auto predResult = CompareTensors(result.output, result.outputExpected);
+    auto predResult = CompareTensors(result.m_ActualData,  result.m_ExpectedData,
+                                     result.m_ActualShape, result.m_ExpectedShape);
     BOOST_TEST(predResult.m_Result, predResult.m_Message.str());
 }
 
@@ -43,7 +46,8 @@ BOOST_AUTO_TEST_CASE(CopyBetweenGpuAndCpuWithSubtensors)
 {
     LayerTestResult<float, 4> result =
         MemCopyTest<armnn::ClWorkloadFactory, armnn::RefWorkloadFactory, armnn::DataType::Float32>(true);
-    auto predResult = CompareTensors(result.output, result.outputExpected);
+    auto predResult = CompareTensors(result.m_ActualData,  result.m_ExpectedData,
+                                     result.m_ActualShape, result.m_ExpectedShape);
     BOOST_TEST(predResult.m_Result, predResult.m_Message.str());
 }