From 4f5e85735b47e0e1625977667e8253a4e676e1af Mon Sep 17 00:00:00 2001
From: Maneesh Gupta <maneesh.gupta@amd.com>
Date: Mon, 10 Oct 2022 20:54:27 +0530
Subject: [PATCH] SWDEV-1 - Add missing checks in hipMemcpy2DToArrayAsync.cc

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

diff --git a/catch/unit/memory/hipMemcpy2DToArrayAsync.cc b/catch/unit/memory/hipMemcpy2DToArrayAsync.cc
index 5145f28aa1..990b9c651e 100644
--- a/catch/unit/memory/hipMemcpy2DToArrayAsync.cc
+++ b/catch/unit/memory/hipMemcpy2DToArrayAsync.cc
@@ -216,7 +216,7 @@ TEST_CASE("Unit_hipMemcpy2DToArrayAsync_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};
@@ -278,7 +278,7 @@ TEST_CASE("Unit_hipMemcpy2DToArrayAsync_multiDeviceDeviceContextChange") {
   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};