From 81051f3520904342e220b269f0e015923f726073 Mon Sep 17 00:00:00 2001
From: Rahul Manocha <rmanocha@amd.com>
Date: Tue, 28 Jan 2025 12:09:32 -0800
Subject: [PATCH] SWDEV-511855 - Fix hipMemcpyPeer to support stream capture
 checks

Change-Id: I7797f069b3ed4240b6785e82da7494a97b4843c6
---
 hipamd/src/hip_peer.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/hipamd/src/hip_peer.cpp b/hipamd/src/hip_peer.cpp
index e309b73129..b43c450530 100644
--- a/hipamd/src/hip_peer.cpp
+++ b/hipamd/src/hip_peer.cpp
@@ -189,7 +189,7 @@ hipError_t hipDeviceEnablePeerAccess(int peerDeviceId, unsigned int flags) {
 hipError_t hipMemcpyPeer(void* dst, int dstDevice, const void* src, int srcDevice,
                          size_t sizeBytes) {
   HIP_INIT_API(hipMemcpyPeer, dst, dstDevice, src, srcDevice, sizeBytes);
-
+  CHECK_STREAM_CAPTURING();
   if (srcDevice >= static_cast<int>(g_devices.size()) ||
       dstDevice >= static_cast<int>(g_devices.size()) ||
       srcDevice < 0 || dstDevice < 0) {