Syncronizing on handle.get_stream() stream as uniform is executed in …

…that stream
rapidsai · May 11, 2022 · e1673a9 · e1673a9
1 parent 3a431f1
commit e1673a9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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);