From a685ab2602a666fa9f104c392f56449d3b392700 Mon Sep 17 00:00:00 2001
From: "yudong.cai" <yudong.cai@zilliz.com>
Date: Thu, 28 Nov 2019 11:27:57 +0800
Subject: [PATCH] #527 update ground truth check

---
 .../unittest/faiss_benchmark/faiss_benchmark_test.cpp      | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp b/core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp
index 9c2a072390..3c9b3a375c 100644
--- a/core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp
+++ b/core/src/index/unittest/faiss_benchmark/faiss_benchmark_test.cpp
@@ -35,13 +35,13 @@
 #include <faiss/gpu/StandardGpuResources.h>
 #include <faiss/index_io.h>
 
-#ifdef USE_FAISS_V_0_3_0 // faiss_0.3.0
+#ifdef USE_FAISS_V_0_3_0  // faiss_0.3.0
 
 #include <faiss/gpu/GpuCloner.h>
 #include <faiss/index_factory.h>
 #include <faiss/utils/distances.h>
 
-#else // faiss_0.2.1
+#else  // faiss_0.2.1
 
 #include <faiss/gpu/GpuAutoTune.h>
 #include <faiss/utils.h>
@@ -210,7 +210,8 @@ GetResultHitCount(const faiss::Index::idx_t* ground_index, const faiss::Index::i
     size_t min_k = std::min(ground_k, k);
     int hit = 0;
     for (int i = 0; i < nq; i++) {
-        std::set<faiss::Index::idx_t> ground(ground_index + i * ground_k, ground_index + i * ground_k + min_k);
+        std::set<faiss::Index::idx_t> ground(ground_index + i * ground_k,
+                                             ground_index + i * ground_k + min_k / index_add_loops);
         for (int j = 0; j < min_k; j++) {
             faiss::Index::idx_t id = index[i * k + j];
             if (ground.count(id) > 0) {