From e1673a9e3acb30c6d3be2ccb826ac7c466e94425 Mon Sep 17 00:00:00 2001
From: Vinay D <vinayd@nvidia.com>
Date: Wed, 11 May 2022 23:06:35 +0530
Subject: [PATCH 1/2] Syncronizing on handle.get_stream() stream as uniform is
 executed in that stream

---
 cpp/test/linalg/divide.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/cpp/test/linalg/divide.cu b/cpp/test/linalg/divide.cu
index 914ef21269..96fdba820b 100644
--- a/cpp/test/linalg/divide.cu
+++ b/cpp/test/linalg/divide.cu
@@ -59,6 +59,7 @@ class DivideTest : public ::testing::TestWithParam<raft::linalg::UnaryOpInputs<T
     int len = params.len;
     RAFT_CUDA_TRY(cudaStreamCreate(&stream));
     uniform(handle, r, in.data(), len, T(-1.0), T(1.0));
+    RAFT_CUDA_TRY(cudaStreamSynchronize(handle.get_stream()));
     naiveDivide(out_ref.data(), in.data(), params.scalar, len, stream);
     divideScalar(out.data(), in.data(), params.scalar, len, stream);
     handle.sync_stream(stream);

From 5c9c9a44a02572df97035b55ca6e5e6f9623f6ca Mon Sep 17 00:00:00 2001
From: "Corey J. Nolet" <cjnolet@gmail.com>
Date: Wed, 11 May 2022 16:47:22 -0400
Subject: [PATCH 2/2] Removing explicit stream create

---
 cpp/test/linalg/divide.cu | 2 --
 1 file changed, 2 deletions(-)

diff --git a/cpp/test/linalg/divide.cu b/cpp/test/linalg/divide.cu
index 96fdba820b..d620979c2f 100644
--- a/cpp/test/linalg/divide.cu
+++ b/cpp/test/linalg/divide.cu
@@ -57,9 +57,7 @@ class DivideTest : public ::testing::TestWithParam<raft::linalg::UnaryOpInputs<T
   {
     raft::random::RngState r(params.seed);
     int len = params.len;
-    RAFT_CUDA_TRY(cudaStreamCreate(&stream));
     uniform(handle, r, in.data(), len, T(-1.0), T(1.0));
-    RAFT_CUDA_TRY(cudaStreamSynchronize(handle.get_stream()));
     naiveDivide(out_ref.data(), in.data(), params.scalar, len, stream);
     divideScalar(out.data(), in.data(), params.scalar, len, stream);
     handle.sync_stream(stream);