From 74431ddfd8df2d0e06a656b16933773eb5792996 Mon Sep 17 00:00:00 2001 From: "Corey J. Nolet" Date: Fri, 10 Feb 2023 19:31:34 -0500 Subject: [PATCH] Update cpp/src/kmeans/kmeans_mg_impl.cuh Co-authored-by: Divye Gala --- cpp/src/kmeans/kmeans_mg_impl.cuh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cpp/src/kmeans/kmeans_mg_impl.cuh b/cpp/src/kmeans/kmeans_mg_impl.cuh index 0725d57763..cb7ead8bf7 100644 --- 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(n_clusters), newCentroids.data_handle(), stream);