diff --git a/cpp/include/raft/sparse/solver/detail/mst_solver_inl.cuh b/cpp/include/raft/sparse/solver/detail/mst_solver_inl.cuh index cbe3ed480d..b7b3b9de45 100644 --- a/cpp/include/raft/sparse/solver/detail/mst_solver_inl.cuh +++ b/cpp/include/raft/sparse/solver/detail/mst_solver_inl.cuh @@ -225,6 +225,7 @@ void MST_solver::alteration() // Random number generator curandGenerator_t randGen; curandCreateGenerator(&randGen, CURAND_RNG_PSEUDO_DEFAULT); + curandSetStream(randGen, stream); curandSetPseudoRandomGeneratorSeed(randGen, 1234567); // Initialize rand values