Commit 4b8a72f9 authored by 余昆's avatar 余昆
Browse files

NSG build failed using GPU-edition if set gpu_enable false

parent 018abff2

CHANGELOG.md

+1 −0
Original line number Diff line number Diff line
@@ -33,6 +33,7 @@ Please mark all change in change log and use the ticket from JIRA. 
- \#533 - NSG build failed with MetricType Inner Product

- \#543 - client raise exception in shards when search results is empty

- \#545 - Avoid dead circle of build index thread when error occurs

- \#547 - NSG build failed using GPU-edition if set gpu_enable false

- \#552 - Server down during building index_type: IVF_PQ using GPU-edition

- \#561 - Milvus server should report exception/error message or terminate on mysql metadata backend error

- \#599 - Build index log is incorrect

core/src/index/knowhere/knowhere/index/vector_index/IndexNSG.cpp

+6 −0
Original line number Diff line number Diff line
@@ -117,7 +117,13 @@ NSG::Train(const DatasetPtr& dataset, const Config& config) { 


    // TODO(linxj): dev IndexFactory, support more IndexType

#ifdef MILVUS_GPU_VERSION

    auto temp_resource = FaissGpuResourceMgr::GetInstance().GetRes(build_cfg->gpu_id);

#if temp_resource == nullptr

    auto preprocess_index = std::make_shared<IVF>();

#else

    auto preprocess_index = std::make_shared<GPUIVF>(build_cfg->gpu_id);

#endif



#else

    auto preprocess_index = std::make_shared<IVF>();

#endif