Fixign a couple small kmeans bugs (#1274)

Authors:
  - Corey J. Nolet (https://github.com/cjnolet)

Approvers:
  - Divye Gala (https://github.com/divyegala)
  - Ben Frederickson (https://github.com/benfred)

URL: #1274

cpp/include/raft/cluster/detail/kmeans.cuh

@@ -1209,7 +1209,7 @@ void kmeans_transform(raft::device_resources const& handle,
   // n_clusters]
   for (IndexT dIdx = 0; dIdx < (IndexT)n_samples; dIdx += dataBatchSize) {
     // # of samples for the current batch
-    auto ns = std::min(dataBatchSize, n_samples - dIdx);
+    auto ns = std::min(static_cast<IndexT>(dataBatchSize), static_cast<IndexT>(n_samples - dIdx));
 
     // datasetView [ns x n_features] - view representing the current batch of
     // input dataset

cpp/include/raft/random/detail/make_blobs.cuh

@@ -158,9 +158,9 @@ void generate_data(DataT* out,
                    raft::random::RngState& rng_state)
 {
   constexpr IdxT block_size = 128;
-  IdxT items                = n_rows * n_cols;
+  int64_t items             = static_cast<int64_t>(n_rows) * n_cols;
   // Choose a grid size so that each thread can write two output values.
-  IdxT nBlocks = ceildiv<IdxT>(items, 2 * block_size);
+  int64_t nBlocks = ceildiv<int64_t>(items, 2 * block_size);
   // parentheses needed here for kernel, otherwise macro interprets the arguments
   // of triple chevron notation as macro arguments
   RAFT_CALL_RNG_FUNC(rng_state,