Fix issue with limit resource adapter

cpp/src/umap/knn_graph/algo.cuh

@@ -100,14 +100,17 @@ inline void launcher(const raft::handle_t& handle,
                      const ML::UMAPParams* params,
                      cudaStream_t stream)
 {
-  size_t free_mem, total_mem;
-  cudaMemGetInfo(&free_mem, &total_mem);
-  double factor = 4.0;
+  raft::resources tmp_handle(handle);
+  raft::resource::set_workspace_resource(tmp_handle);
+  size_t free_size = raft::resource::get_workspace_free_bytes(tmp_handle);
+
+  double factor           = 4.0;
   size_t index_batch_size = inputsA.n;
   size_t query_batch_size = inputsB.n;
-  size_t requirements = factor * sizeof(float) * index_batch_size * query_batch_size;
-  if (requirements > free_mem){
-    index_batch_size = free_mem / (query_batch_size * factor * sizeof(float));
+  size_t requirements     = factor * sizeof(float) * index_batch_size * query_batch_size;
+
+  if (requirements > free_size) {
+    index_batch_size = free_size / (query_batch_size * factor * sizeof(float));
   }
 
   raft::sparse::selection::brute_force_knn(inputsA.indptr,