Fix PR problems (facebookresearch#2839)

Summary: Pull Request resolved: facebookresearch#2839 Reviewed By: algoriddle Differential Revision: D45054275 fbshipit-source-id: 7eca136a6baff97e244d570c3ba7ab3171ccca48
blevesearch · Sep 26, 2023 · 8568ef8 · 8568ef8
1 parent e863178
commit 8568ef8
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 13 deletions.
diff --git a/faiss/gpu/GpuDistance.cu b/faiss/gpu/GpuDistance.cu
@@ -293,7 +293,6 @@ void bfKnn(GpuResourcesProvider* prov, const GpuDistanceParams& args) {
                         search.view(),
                         inds.view(),
                         dists.view(),
-                        k,
                         distance,
                         metric_arg);
             }
@@ -328,7 +327,6 @@ void bfKnn(GpuResourcesProvider* prov, const GpuDistanceParams& args) {
                     search.view(),
                     inds.view(),
                     dists.view(),
-                    k,
                     distance,
                     metric_arg);
         }

diff --git a/faiss/gpu/StandardGpuResources.cpp b/faiss/gpu/StandardGpuResources.cpp
@@ -89,22 +89,21 @@ std::string allocsToString(const std::unordered_map<void*, AllocRequest>& map) {
 //
 
 StandardGpuResourcesImpl::StandardGpuResourcesImpl()
-        : pinnedMemAlloc_(nullptr),
+        :
+#if defined USE_NVIDIA_RAFT
+          cmr(new rmm::mr::cuda_memory_resource),
+          mmr(new rmm::mr::managed_memory_resource),
+          pmr(new rmm::mr::pinned_memory_resource),
+#endif
+          pinnedMemAlloc_(nullptr),
           pinnedMemAllocSize_(0),
           // let the adjustment function determine the memory size for us by
           // passing in a huge value that will then be adjusted
           tempMemSize_(getDefaultTempMemForGPU(
                   -1,
                   std::numeric_limits<size_t>::max())),
           pinnedMemSize_(kDefaultPinnedMemoryAllocation),
-          allocLogging_(false)
-#if defined USE_NVIDIA_RAFT
-          ,
-          cmr(new rmm::mr::cuda_memory_resource),
-          mmr(new rmm::mr::managed_memory_resource),
-          pmr(new rmm::mr::pinned_memory_resource)
-#endif
-{
+          allocLogging_(false) {
 }
 
 StandardGpuResourcesImpl::~StandardGpuResourcesImpl() {

diff --git a/faiss/gpu/impl/RaftFlatIndex.cu b/faiss/gpu/impl/RaftFlatIndex.cu
@@ -109,7 +109,6 @@ void RaftFlatIndex::query(
                     search,
                     inds,
                     dists,
-                    k,
                     distance,
                     metricArg);
         }

diff --git a/faiss/gpu/impl/RaftUtils.h b/faiss/gpu/impl/RaftUtils.h
@@ -20,13 +20,16 @@
  * limitations under the License.
  */
 
+#pragma once
+
 #include <faiss/MetricType.h>
+#include <raft/core/error.hpp>
 #include <raft/distance/distance_types.hpp>
 
 namespace faiss {
 namespace gpu {
 
-raft::distance::DistanceType faiss_to_raft(
+inline raft::distance::DistanceType faiss_to_raft(
         MetricType metric,
         bool exactDistance) {
     switch (metric) {