Update cpp/src/glm/qn_mg.cu

Co-authored-by: Simon Adorf <sadorf@nvidia.com>
rapidsai · Sep 27, 2023 · 1b4050c · 1b4050c
1 parent e877100
commit 1b4050c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cpp/src/glm/qn_mg.cu b/cpp/src/glm/qn_mg.cu
@@ -53,7 +53,8 @@ std::vector<T> distinct_mg(const raft::handle_t& handle, T* y, size_t n)
   raft::copy(recv_counts_host.data(), recv_counts.data(), n_ranks, stream);
 
   std::vector<size_t> displs(n_ranks);
-  for (int i = 0, pos = 0; i < n_ranks; ++i) {
+  size_t pos = 0;
+  for (int i = 0; i < n_ranks; ++i) {
     displs[i] = pos;
     pos += recv_counts_host[i];
   }