Update cpp/src/kmeans/kmeans_mg_impl.cuh

Co-authored-by: Divye Gala <[email protected]>
rapidsai · Feb 13, 2023 · 74431dd · 74431dd
1 parent 9e2ea9f
commit 74431dd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cpp/src/kmeans/kmeans_mg_impl.cuh b/cpp/src/kmeans/kmeans_mg_impl.cuh
@@ -579,7 +579,7 @@ void fit(const raft::handle_t& handle,
                                      workspace.data(),
                                      X.extent(0),
                                      X.extent(1),
-                                     (IndexT)n_clusters,
+                                     static_cast<IndexT>(n_clusters),
                                      newCentroids.data_handle(),
                                      stream);