From f14ad3abfefc20bbc339fb604deccb798612a67d Mon Sep 17 00:00:00 2001
From: Maneesh Gupta <maneesh.gupta@amd.com>
Date: Mon, 10 Oct 2022 16:14:04 +0530
Subject: [PATCH] SWDEV-1 - Add missing checks in hipMemcpy2DToArray.cc

[ROCm/hip commit: 86d99b5ee6a84f3431f4074b000225a72beb0fea]
---
 projects/hip/tests/catch/unit/memory/hipMemcpy2DToArray.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/projects/hip/tests/catch/unit/memory/hipMemcpy2DToArray.cc b/projects/hip/tests/catch/unit/memory/hipMemcpy2DToArray.cc
index dc1cf73348..e75e2c6af3 100644
--- a/projects/hip/tests/catch/unit/memory/hipMemcpy2DToArray.cc
+++ b/projects/hip/tests/catch/unit/memory/hipMemcpy2DToArray.cc
@@ -191,7 +191,7 @@ TEST_CASE("Unit_hipMemcpy2DToArray_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};
@@ -248,7 +248,7 @@ TEST_CASE("Unit_hipMemcpy2DToArray_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};