Skip to content

Commit

Permalink
try to fix style
Browse files Browse the repository at this point in the history
  • Loading branch information
divyegala committed Oct 6, 2023
1 parent e2da531 commit 5322f7c
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 9 deletions.
6 changes: 3 additions & 3 deletions cpp/include/raft/stats/detail/neighborhood_recall.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -48,8 +48,8 @@ __global__ void neighborhood_recall(
DistanceValueType const eps)
{
auto constexpr kThreadsPerBlock = 32;
IndexType const row_idx = blockIdx.x;
auto const lane_idx = threadIdx.x % kThreadsPerBlock;
IndexType const row_idx = blockIdx.x;
auto const lane_idx = threadIdx.x % kThreadsPerBlock;

// Each warp stores a recall score computed across the columns per row
IndexType thread_recall_score = 0;
Expand Down Expand Up @@ -106,7 +106,7 @@ void neighborhood_recall(
{
// One warp per row, launch a warp-width block per-row kernel
auto constexpr kThreadsPerBlock = 32;
auto const num_blocks = indices.extent(0);
auto const num_blocks = indices.extent(0);

neighborhood_recall<<<num_blocks, kThreadsPerBlock>>>(
indices, ref_indices, distances, ref_distances, recall_score, eps);
Expand Down
12 changes: 6 additions & 6 deletions cpp/include/raft/stats/neighborhood_recall.cuh
Original file line number Diff line number Diff line change
Expand Up @@ -60,10 +60,10 @@ namespace raft::stats {
* auto recall_score = raft::make_device_scalar(res, scalar);
*
* raft::stats::neighborhood_recall(res,
raft::make_const_mdspan(indices.view()),
raft::make_const_mdspan(ref_indices.view()),
raft::make_const_mdspan(indices.view()),
raft::make_const_mdspan(ref_indices.view()),
recall_score.view(),
raft::make_const_mdspan(distances.view()),
raft::make_const_mdspan(distances.view()),
raft::make_const_mdspan(ref_distances.view()));
* @endcode
*
Expand Down Expand Up @@ -145,10 +145,10 @@ void neighborhood_recall(
* auto recall_score = raft::make_host_scalar(scalar);
*
* raft::stats::neighborhood_recall(res,
raft::make_const_mdspan(indices.view()),
raft::make_const_mdspan(ref_indices.view()),
raft::make_const_mdspan(indices.view()),
raft::make_const_mdspan(ref_indices.view()),
recall_score.view(),
raft::make_const_mdspan(distances.view()),
raft::make_const_mdspan(distances.view()),
raft::make_const_mdspan(ref_distances.view()));
* @endcode
*
Expand Down

0 comments on commit 5322f7c

Please sign in to comment.