diff --git a/cpp/include/raft/stats/detail/neighborhood_recall.cuh b/cpp/include/raft/stats/detail/neighborhood_recall.cuh index 11d0448167..3f05225e74 100644 --- a/cpp/include/raft/stats/detail/neighborhood_recall.cuh +++ b/cpp/include/raft/stats/detail/neighborhood_recall.cuh @@ -1,5 +1,5 @@ /* - * Copyright (c) 2023, NVIDIA CORPORATION. + * Copyright (c) 2023-2024, NVIDIA CORPORATION. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. @@ -23,6 +23,7 @@ #include #include #include +#include #include #include @@ -108,7 +109,7 @@ void neighborhood_recall( auto constexpr kThreadsPerBlock = 32; auto const num_blocks = indices.extent(0); - neighborhood_recall<<>>( + neighborhood_recall<<>>( indices, ref_indices, distances, ref_distances, recall_score, eps); }