diff --git a/cpp/include/raft/neighbors/detail/nn_descent.cuh b/cpp/include/raft/neighbors/detail/nn_descent.cuh index 009ffd4684..1fb568a934 100644 --- a/cpp/include/raft/neighbors/detail/nn_descent.cuh +++ b/cpp/include/raft/neighbors/detail/nn_descent.cuh @@ -980,9 +980,9 @@ GnndGraph::GnndGraph(const size_t nrow, num_samples(num_samples), bloom_filter(nrow, internal_node_degree / segment_size, 3), h_dists{raft::make_host_matrix(nrow, node_degree)}, - h_graph_new{nrow * num_samples}, - h_list_sizes_new{nrow}, - h_graph_old{nrow * num_samples}, + h_graph_new(nrow * num_samples), + h_list_sizes_new(nrow), + h_graph_old(nrow * num_samples), h_list_sizes_old{nrow} { // node_degree must be a multiple of segment_size; @@ -1150,12 +1150,12 @@ GNND::GNND(raft::resources const& res, const BuildConfig& build raft::make_device_matrix(res, nrow_, DEGREE_ON_DEVICE)}, dists_buffer_{ raft::make_device_matrix(res, nrow_, DEGREE_ON_DEVICE)}, - graph_host_buffer_{static_cast(nrow_ * DEGREE_ON_DEVICE)}, - dists_host_buffer_{static_cast(nrow_ * DEGREE_ON_DEVICE)}, + graph_host_buffer_(nrow_ * DEGREE_ON_DEVICE), + dists_host_buffer_(nrow_ * DEGREE_ON_DEVICE), d_locks_{raft::make_device_vector(res, nrow_)}, - h_rev_graph_new_{static_cast(nrow_ * NUM_SAMPLES)}, - h_graph_old_{static_cast(nrow_ * NUM_SAMPLES)}, - h_rev_graph_old_{static_cast(nrow_ * NUM_SAMPLES)}, + h_rev_graph_new_(nrow_ * NUM_SAMPLES), + h_graph_old_(nrow_ * NUM_SAMPLES), + h_rev_graph_old_(nrow_ * NUM_SAMPLES), d_list_sizes_new_{raft::make_device_vector(res, nrow_)}, d_list_sizes_old_{raft::make_device_vector(res, nrow_)} {