From 59ace1e08078fafc5b7c8afb2caf80950e1e8d60 Mon Sep 17 00:00:00 2001
From: Vladana Stojiljkovic <vladana.stojiljkovic@amd.com>
Date: Mon, 25 Nov 2024 11:42:02 +0200
Subject: [PATCH] SWDEV-491280 - Create capture tests for Peer APIs

Change-Id: Ib35ab1155b19e00f7f297e6eaf395709d5767d7c
---
 catch/unit/device/hipDeviceCanAccessPeer.cc           | 7 +++++++
 catch/unit/device/hipDeviceEnableDisablePeerAccess.cc | 8 ++++++++
 2 files changed, 15 insertions(+)

diff --git a/catch/unit/device/hipDeviceCanAccessPeer.cc b/catch/unit/device/hipDeviceCanAccessPeer.cc
index 6ef6ca5926..98e52acd5c 100644
--- a/catch/unit/device/hipDeviceCanAccessPeer.cc
+++ b/catch/unit/device/hipDeviceCanAccessPeer.cc
@@ -54,7 +54,14 @@ TEST_CASE("Unit_hipDeviceCanAccessPeer_positive") {
   int dev = GENERATE(range(0, HipTest::getGeviceCount()));
   int peerDev = GENERATE(range(0, HipTest::getGeviceCount()));
 
+  GENERATE_CAPTURE();
+  hipStream_t stream;
+  HIP_CHECK(hipStreamCreate(&stream));
+  BEGIN_CAPTURE(stream);
   HIP_CHECK(hipDeviceCanAccessPeer(&canAccessPeer, dev, peerDev));
+  END_CAPTURE(stream);
+  HIP_CHECK(hipStreamDestroy(stream));
+
   if (dev != peerDev) {
     REQUIRE(canAccessPeer >= 0);
   }
diff --git a/catch/unit/device/hipDeviceEnableDisablePeerAccess.cc b/catch/unit/device/hipDeviceEnableDisablePeerAccess.cc
index bf85cbe0e8..c97318d894 100644
--- a/catch/unit/device/hipDeviceEnableDisablePeerAccess.cc
+++ b/catch/unit/device/hipDeviceEnableDisablePeerAccess.cc
@@ -59,7 +59,15 @@ TEST_CASE("Unit_hipDeviceEnableDisablePeerAccess_positive") {
 
   if (dev != peerDev) {
     HIP_CHECK(hipSetDevice(dev));
+
+    GENERATE_CAPTURE();
+    hipStream_t stream;
+    HIP_CHECK(hipStreamCreate(&stream));
+    BEGIN_CAPTURE(stream);
     HIP_CHECK(hipDeviceCanAccessPeer(&canAccessPeer, dev, peerDev));
+    END_CAPTURE(stream);
+    HIP_CHECK(hipStreamDestroy(stream));
+
     if (canAccessPeer == 0) {
       HipTest::HIP_SKIP_TEST("Skipping because no P2P support");
       return;