From 4cbfbe211229c874ef5bce27a5d311f13ffb399a 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


[ROCm/clr commit: 81051f3520904342e220b269f0e015923f726073]
---
 projects/clr/hipamd/src/hip_peer.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/projects/clr/hipamd/src/hip_peer.cpp b/projects/clr/hipamd/src/hip_peer.cpp
index e309b73129..b43c450530 100644
--- a/projects/clr/hipamd/src/hip_peer.cpp
+++ b/projects/clr/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) {