From eb894a8b8d523907bf9de72d84c55a6e213e8cb8 Mon Sep 17 00:00:00 2001
From: Jaydeep Patel <jaydeepkumar.patel@amd.com>
Date: Thu, 14 Nov 2024 11:48:42 +0000
Subject: [PATCH] SWDEV-497532 - Pass correct stream to
 MemcpyDeviceToDeviceShell to sync.

Change-Id: I7fe2b39e49725d01e7d9ec5b4ab9927a96dc951c
---
 catch/unit/memory/hipMemcpyAsync_derivatives.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/catch/unit/memory/hipMemcpyAsync_derivatives.cc b/catch/unit/memory/hipMemcpyAsync_derivatives.cc
index b2583d1ccc..1fabfd3785 100644
--- a/catch/unit/memory/hipMemcpyAsync_derivatives.cc
+++ b/catch/unit/memory/hipMemcpyAsync_derivatives.cc
@@ -133,14 +133,14 @@ TEST_CASE("Unit_hipMemcpyDtoDAsync_Positive_Basic") {
           [stream = stream_guard.stream()](void* dst, void* src, size_t count) {
             return hipMemcpyDtoDAsync(reinterpret_cast<hipDeviceptr_t>(dst),
                                       reinterpret_cast<hipDeviceptr_t>(src), count, stream);
-          });
+          }, stream_guard.stream());
     }
     SECTION("Peer access disabled") {
       MemcpyDeviceToDeviceShell<true, false>(
           [stream = stream_guard.stream()](void* dst, void* src, size_t count) {
             return hipMemcpyDtoDAsync(reinterpret_cast<hipDeviceptr_t>(dst),
                                       reinterpret_cast<hipDeviceptr_t>(src), count, stream);
-          });
+          }, stream_guard.stream());
     }
   }
 }