From e99cd4b50a80b6602b30d8c274cf1024bbb95204 Mon Sep 17 00:00:00 2001
From: Maneesh Gupta <maneesh.gupta@amd.com>
Date: Tue, 11 Oct 2022 08:29:39 +0530
Subject: [PATCH] SWDEV-1 - Add missing checks in hipMemcpy2DFromArray.cc

---
 tests/catch/unit/memory/hipMemcpy2DFromArray.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/catch/unit/memory/hipMemcpy2DFromArray.cc b/tests/catch/unit/memory/hipMemcpy2DFromArray.cc
index 0e6a7aacde..1972a5f10a 100644
--- a/tests/catch/unit/memory/hipMemcpy2DFromArray.cc
+++ b/tests/catch/unit/memory/hipMemcpy2DFromArray.cc
@@ -190,7 +190,7 @@ TEST_CASE("Unit_hipMemcpy2DFromArray_multiDevicePinnedMemPeerGpu") {
   HIP_CHECK(hipGetDeviceCount(&numDevices));
   if (numDevices > 1) {
     int canAccessPeer = 0;
-    hipDeviceCanAccessPeer(&canAccessPeer, 0, 1);
+    HIP_CHECK(hipDeviceCanAccessPeer(&canAccessPeer, 0, 1));
     if (canAccessPeer) {
       HIP_CHECK(hipSetDevice(0));
       hipArray *A_d{nullptr};
@@ -246,7 +246,7 @@ TEST_CASE("Unit_hipMemcpy2DFromArray_multiDeviceContextChange") {
   HIP_CHECK(hipGetDeviceCount(&numDevices));
   if (numDevices > 1) {
     int canAccessPeer = 0;
-    hipDeviceCanAccessPeer(&canAccessPeer, 0, 1);
+    HIP_CHECK(hipDeviceCanAccessPeer(&canAccessPeer, 0, 1));
     if (canAccessPeer) {
       HIP_CHECK(hipSetDevice(0));
       hipArray *A_d{nullptr};