Fix gpu compile error (#196)

Signed-off-by: Yusheng.Ma <[email protected]>
milvus-io · Jun 2, 2022 · a40166e · a40166e
1 parent abc1f53
commit a40166e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/knowhere/index/VecIndexFactory.cpp b/knowhere/index/VecIndexFactory.cpp
@@ -108,7 +108,7 @@ VecIndexFactory::CreateVecIndex(const IndexType& type, const IndexMode mode) {
             } else if (type == IndexEnum::INDEX_FAISS_IDMAP) {
                 return std::make_shared<knowhere::GPUIDMAP>(gpu_device);
             } else if (type == IndexEnum::INDEX_FAISS_IVFFLAT) {
-                return std::make_shared<knowhere::GPUIVF_NM>(gpu_device);
+                return std::make_shared<knowhere::GPUIVF>(gpu_device);
             } else if (type == IndexEnum::INDEX_FAISS_IVFPQ) {
                 return std::make_shared<knowhere::GPUIVFPQ>(gpu_device);
             } else if (type == IndexEnum::INDEX_FAISS_IVFSQ8) {

diff --git a/knowhere/index/vector_offset_index/gpu/IndexGPUIVF_NM.h b/knowhere/index/vector_offset_index/gpu/IndexGPUIVF_NM.h
@@ -36,9 +36,6 @@ class GPUIVF_NM : public IVF, public GPUIndex {
     void
     AddWithoutIds(const DatasetPtr&, const Config&) override;
 
-    void
-    Load(const BinarySet&) override;
-
     VecIndexPtr
     CopyGpuToCpu(const Config&) override;
 
@@ -55,6 +52,9 @@ class GPUIVF_NM : public IVF, public GPUIndex {
     BinarySet
     SerializeImpl(const IndexType&) override;
 
+    void
+    LoadImpl(const BinarySet&, const IndexType&);
+
     void
     QueryImpl(int64_t, const float*, int64_t, float*, int64_t*, const Config&, const faiss::BitsetView) override;