Merge branch 'main' into fea-dry-run-protocol

Author: achirkin

cpp/include/raft/comms/detail/test.hpp

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION.
+ * SPDX-FileCopyrightText: Copyright (c) 2021-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
 
@@ -376,8 +376,9 @@ bool test_pointToPoint_device_send_or_recv(raft::resources const& h, int numTria
       std::cout << "Trial " << i << std::endl;
     }
 
-    bool sender = (rank % 2) == 0 ? true : false;
-    rmm::device_scalar<int> received_data(-1, stream);
+    bool sender             = (rank % 2) == 0 ? true : false;
+    int received_data_value = -1;
+    rmm::device_scalar<int> received_data(received_data_value, stream);
     rmm::device_scalar<int> sent_data(rank, stream);
 
     if (sender) {
@@ -418,7 +419,8 @@ bool test_pointToPoint_device_sendrecv(raft::resources const& h, int numTrials)
       std::cout << "Trial " << i << std::endl;
     }
 
-    rmm::device_scalar<int> received_data(-1, stream);
+    int received_data_value = -1;
+    rmm::device_scalar<int> received_data(received_data_value, stream);
     rmm::device_scalar<int> sent_data(rank, stream);
 
     if (rank % 2 == 0) {

cpp/include/raft/sparse/solver/detail/mst_solver_inl.cuh

@@ -80,8 +80,8 @@ MST_solver<vertex_t, edge_t, weight_t, alteration_t>::MST_solver(raft::resources
     temp_src(2 * v_, stream_),
     temp_dst(2 * v_, stream_),
     temp_weights(2 * v_, stream_),
-    mst_edge_count(1, stream_),
-    prev_mst_edge_count(1, stream_),
+    mst_edge_count(stream_),
+    prev_mst_edge_count(stream_),
     stream(stream_),
     symmetrize_output(symmetrize_output_),
     initialize_colors(initialize_colors_),
@@ -225,6 +225,7 @@ void MST_solver<vertex_t, edge_t, weight_t, alteration_t>::alteration()
   // Random number generator
   curandGenerator_t randGen;
   curandCreateGenerator(&randGen, CURAND_RNG_PSEUDO_DEFAULT);
+  curandSetStream(randGen, stream);
   curandSetPseudoRandomGeneratorSeed(randGen, 1234567);
 
   // Initialize rand values

cpp/tests/linalg/dot.cu

@@ -63,7 +63,8 @@ class DotTest : public ::testing::TestWithParam<DotInputs<T>> {
     uniform(handle, r, x.data(), x_len, T(-1.0), T(1.0));
     uniform(handle, r, y.data(), y_len, T(-1.0), T(1.0));
 
-    rmm::device_scalar<T> ref(0, resource::get_cuda_stream(handle));
+    T zero = 0;
+    rmm::device_scalar<T> ref(zero, resource::get_cuda_stream(handle));
     raft::launch_kernel(handle,
                         256,
                         256,
@@ -77,7 +78,7 @@ class DotTest : public ::testing::TestWithParam<DotInputs<T>> {
     raft::update_host(&ref_output, ref.data(), 1, stream);
 
     // Test out both the device and host api's
-    rmm::device_scalar<T> out(0, resource::get_cuda_stream(handle));
+    rmm::device_scalar<T> out(zero, resource::get_cuda_stream(handle));
     auto device_out_view = make_device_scalar_view<T, IndexType>(out.data());
     auto host_out_view   = make_host_scalar_view<T, IndexType>(&host_output);
 

cpp/tests/linalg/mean_squared_error.cu

@@ -43,14 +43,15 @@ class MeanSquaredErrorTest : public ::testing::TestWithParam<MeanSquaredErrorInp
   MeanSquaredErrorInputs<T> params;
 
   raft::resources handle;
+  T zero = 0;
   rmm::device_scalar<T> output;
   rmm::device_scalar<T> refoutput;
 
  public:
   MeanSquaredErrorTest()
     : testing::TestWithParam<MeanSquaredErrorInputs<T>>(),
-      output(0, resource::get_cuda_stream(handle)),
-      refoutput(0, resource::get_cuda_stream(handle))
+      output(zero, resource::get_cuda_stream(handle)),
+      refoutput(zero, resource::get_cuda_stream(handle))
   {
     resource::sync_stream(handle);
   }

cpp/tests/util/device_atomics.cu

@@ -38,7 +38,8 @@ TEST(Raft, AtomicIncWarp)
   rmm::cuda_stream_pool pool{1};
   auto s = pool.get_stream();
 
-  rmm::device_scalar<int> counter{0, s};
+  int zero = 0;
+  rmm::device_scalar<int> counter{zero, s};
   rmm::device_uvector<int> out_device{num_elts, s};
   std::array<int, num_elts> out_host{0};
 

cpp/tests/util/popc.cu

@@ -93,7 +93,8 @@ class PopcTest : public ::testing::TestWithParam<PopcInputs<index_t>> {
     auto max_len_view = raft::make_host_scalar_view<const index_t, index_t>(&max_len);
 
     index_t nnz_actual_h = 0;
-    rmm::device_scalar<index_t> nnz_actual_d(0, stream);
+    index_t zero         = 0;
+    rmm::device_scalar<index_t> nnz_actual_d(zero, stream);
     auto nnz_actual_view = raft::make_device_scalar_view<index_t>(nnz_actual_d.data());
 
     raft::execute_with_dry_run_check(