Skip to content

Commit

Permalink
Drop deprecated boolean argument for cub::DeviceSegmentedReduce::Max
Browse files Browse the repository at this point in the history
  • Loading branch information
miscco committed Jan 23, 2025
1 parent a9567b4 commit e5a0f26
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 8 deletions.
5 changes: 2 additions & 3 deletions cpp/src/hdbscan/detail/membership.cuh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright (c) 2021-2024, NVIDIA CORPORATION.
* Copyright (c) 2021-2025, NVIDIA CORPORATION.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -80,8 +80,7 @@ void get_probabilities(const raft::handle_t& handle,
int,
const value_idx*,
const value_idx*,
cudaStream_t,
bool) =
cudaStream_t) =
cub::DeviceSegmentedReduce::Max<const value_t*, value_t*, const value_idx*, const value_idx*>;
Utils::cub_segmented_reduce(
lambdas, deaths.data(), n_clusters, sorted_parents_offsets.data(), stream, reduce_func);
Expand Down
5 changes: 2 additions & 3 deletions cpp/src/hdbscan/detail/stabilities.cuh
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright (c) 2021-2024, NVIDIA CORPORATION.
* Copyright (c) 2021-2025, NVIDIA CORPORATION.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -104,8 +104,7 @@ void compute_stabilities(const raft::handle_t& handle,
int,
const value_idx*,
const value_idx*,
cudaStream_t,
bool) =
cudaStream_t) =
cub::DeviceSegmentedReduce::Min<const value_t*, value_t*, const value_idx*, const value_idx*>;
Utils::cub_segmented_reduce(lambdas,
births_parent_min.data() + 1,
Expand Down
4 changes: 2 additions & 2 deletions cpp/src/hdbscan/prediction_data.cu
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
/*
* Copyright (c) 2022-2024, NVIDIA CORPORATION.
* Copyright (c) 2022-2025, NVIDIA CORPORATION.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
Expand Down Expand Up @@ -130,7 +130,7 @@ void generate_prediction_data(const raft::handle_t& handle,

// this is to find maximum lambdas of all children under a parent
cudaError_t (*reduce_func)(
void*, size_t&, const float*, float*, int, const int*, const int*, cudaStream_t, bool) =
void*, size_t&, const float*, float*, int, const int*, const int*, cudaStream_t) =
cub::DeviceSegmentedReduce::Max<const float*, float*, const int*, const int*>;
detail::Utils::cub_segmented_reduce(lambdas,
prediction_data.get_deaths(),
Expand Down

0 comments on commit e5a0f26

Please sign in to comment.