Commit fd304cf4 authored by 余昆's avatar 余昆
Browse files

remove #if

parent 1b6d90a5

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

+19 −13
Original line number Diff line number Diff line
@@ -116,23 +116,29 @@ NSG::Train(const DatasetPtr& dataset, const Config& config) { 
    }



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

    bool use_gpu = false;

#ifdef MILVUS_GPU_VERSION

    use_gpu = true;

    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);

    if (temp_resource == nullptr)

        use_gpu = false;

#endif

    Graph knng;

    if (use_gpu) {

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

        auto model = preprocess_index->Train(dataset, config);

        preprocess_index->set_index_model(model);

        preprocess_index->AddWithoutIds(dataset, config);



#else

        preprocess_index->GenGraph(build_cfg->knng, knng, dataset, config);

    } else {

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

#endif

        auto model = preprocess_index->Train(dataset, config);

        preprocess_index->set_index_model(model);

        preprocess_index->AddWithoutIds(dataset, config);



    Graph knng;

        preprocess_index->GenGraph(build_cfg->knng, knng, dataset, config);

    }



    algo::BuildParams b_params;

    b_params.candidate_pool_size = build_cfg->candidate_pool_size;