Commit 48ed10ab authored by zirui.chen's avatar zirui.chen
Browse files

fix problem of hardcore nprobe

parent be0bf607

core/src/wrapper/ConfAdapter.cpp

+5 −2
Original line number Diff line number Diff line
@@ -18,6 +18,7 @@ 
#include "wrapper/ConfAdapter.h"

#include "knowhere/index/vector_index/helpers/IndexParameter.h"

#include "utils/Log.h"

#include "WrapperException.h"



#include <cmath>

#include <memory>
@@ -146,8 +147,10 @@ IVFPQConfAdapter::MatchSearch(const TempMetaConf& metaconf, const IndexType& typ 
    auto conf = std::make_shared<knowhere::IVFPQCfg>();

    conf->k = metaconf.k;



    if (metaconf.nprobe <= 0)

        conf->nprobe = 16;  // hardcode here

    if (metaconf.nprobe <= 0){

        WRAPPER_LOG_ERROR << "The nprobe of PQ is wrong!";

        throw WrapperException("The nprobe of PQ is wrong!");

    }

    else

        conf->nprobe = metaconf.nprobe;

core/unittest/wrapper/test_wrapper.cpp

+1 −0
Original line number Diff line number Diff line
@@ -51,6 +51,7 @@ class KnowhereWrapperTest 
        tempconf.size = nb;

        tempconf.dim = dim;

        tempconf.k = k;

        tempconf.nprobe = 16;



        index_ = GetVecIndexFactory(index_type);

        conf = ParamGenerator::GetInstance().GenBuild(index_type, tempconf);