From bb7960222400567bb85c238e8ea05d046bb79a7c Mon Sep 17 00:00:00 2001
From: Maneesh Gupta <maneesh.gupta@amd.com>
Date: Tue, 11 Oct 2022 10:31:40 +0530
Subject: [PATCH] SWDEV-1 - Add missing checks in hipMemcpyPeerAsync.cc

[ROCm/hip commit: a371c151207418f3478f55e73f4a2e951dc31812]
---
 projects/hip/tests/catch/unit/memory/hipMemcpyPeerAsync.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/projects/hip/tests/catch/unit/memory/hipMemcpyPeerAsync.cc b/projects/hip/tests/catch/unit/memory/hipMemcpyPeerAsync.cc
index 1c738f1c76..56a76cbad2 100644
--- a/projects/hip/tests/catch/unit/memory/hipMemcpyPeerAsync.cc
+++ b/projects/hip/tests/catch/unit/memory/hipMemcpyPeerAsync.cc
@@ -39,7 +39,7 @@ TEST_CASE("Unit_hipMemcpyPeerAsync_Negative") {
   HIP_CHECK(hipGetDeviceCount(&numDevices));
   if (numDevices > 1) {
     int canAccessPeer = 0;
-    hipDeviceCanAccessPeer(&canAccessPeer, 0, 1);
+    HIP_CHECK(hipDeviceCanAccessPeer(&canAccessPeer, 0, 1));
     if (canAccessPeer) {
       // Initialization of variables
       int *A_d{nullptr}, *B_d{nullptr};
@@ -116,7 +116,7 @@ TEST_CASE("Unit_hipMemcpyPeerAsync_Basic") {
   HIP_CHECK(hipGetDeviceCount(&numDevices));
   if (numDevices > 1) {
     int canAccessPeer = 0;
-    hipDeviceCanAccessPeer(&canAccessPeer, 0, 1);
+    HIP_CHECK(hipDeviceCanAccessPeer(&canAccessPeer, 0, 1));
     if (canAccessPeer) {
       // Initialization of Variables on GPU-0
       int *A_d{nullptr}, *B_d{nullptr}, *C_d{nullptr};
@@ -202,7 +202,7 @@ TEST_CASE("Unit_hipMemcpyPeerAsync_StreamOnDiffDevice") {
   HIP_CHECK(hipGetDeviceCount(&numDevices));
   if (numDevices > 1) {
     int canAccessPeer = 0;
-    hipDeviceCanAccessPeer(&canAccessPeer, 0, 1);
+    HIP_CHECK(hipDeviceCanAccessPeer(&canAccessPeer, 0, 1));
     if (canAccessPeer) {
       int *A_d{nullptr}, *B_d{nullptr}, *C_d{nullptr};
       int *X_d{nullptr}, *Y_d{nullptr}, *Z_d{nullptr};