From d2767f920cb2e255a61d3ef2b9bd21f3f781aa04 Mon Sep 17 00:00:00 2001
From: dragondriver <jiquan.long@zilliz.com>
Date: Tue, 3 Aug 2021 10:25:25 +0800
Subject: [PATCH] Fix index parameters check of RHNSWPQ (#6950)

Signed-off-by: dragondriver <jiquan.long@zilliz.com>
---
 .../knowhere/knowhere/index/vector_index/ConfAdapter.cpp      | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/internal/core/src/index/knowhere/knowhere/index/vector_index/ConfAdapter.cpp b/internal/core/src/index/knowhere/knowhere/index/vector_index/ConfAdapter.cpp
index a2a77a679c..675df9e0b0 100644
--- 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);
 }