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

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

diff --git a/tests/catch/unit/memory/hipMemcpy2DFromArrayAsync.cc b/tests/catch/unit/memory/hipMemcpy2DFromArrayAsync.cc
index 6603243585..f7c8e75b71 100644
--- a/tests/catch/unit/memory/hipMemcpy2DFromArrayAsync.cc
+++ b/tests/catch/unit/memory/hipMemcpy2DFromArrayAsync.cc
@@ -217,7 +217,7 @@ TEST_CASE("Unit_hipMemcpy2DFromArrayAsync_multiDevicePinnedHostMem") {
   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};
@@ -277,7 +277,7 @@ TEST_CASE("Unit_hipMemcpy2DFromArrayAsync_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};