Fix index parameters check of RHNSWPQ (#6950)

Signed-off-by: N dragondriver <jiquan.long@zilliz.com>

Fix index parameters check of RHNSWPQ (#6950)
Signed-off-by: N dragondriver <jiquan.long@zilliz.com>
d2767f92 · dragondriver · GitHub · 47df4a2e · d2767f92
隐藏空白更改
内联并排

Showing with 3 addition and 1 deletion

internal/core/src/index/knowhere/knowhere/index/vector_index/ConfAdapter.cpp ...ndex/knowhere/knowhere/index/vector_index/ConfAdapter.cpp +3 -1

未找到文件。
--- a/internal/core/src/index/knowhere/knowhere/index/vector_index/ConfAdapter.cpp
+++ b/internal/core/src/index/knowhere/knowhere/index/vector_index/ConfAdapter.cpp
@@ -311,7 +311,9 @@ RHNSWPQConfAdapter::CheckTrain(Config& oricfg, const IndexMode mode) {

    auto dimension = oricfg[knowhere::meta::DIM].get<int64_t>();

-    IVFPQConfAdapter::CheckCPUPQParams(dimension, oricfg[knowhere::IndexParams::PQM].get<int64_t>());
+    if (!IVFPQConfAdapter::CheckCPUPQParams(dimension, oricfg[knowhere::IndexParams::PQM].get<int64_t>())) {
+        return false;
+    }

    return ConfAdapter::CheckTrain(oricfg, mode);
 }