Add STATISTICS_LEVEL for knowhere and faiss (#97)

milvus-io · Mar 18, 2022 · 7e2d03f · 7e2d03f
1 parent a0af665
commit 7e2d03f
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/knowhere/archive/KnowhereConfig.cpp b/knowhere/archive/KnowhereConfig.cpp
@@ -108,9 +108,8 @@ KnowhereConfig::SetClusteringType(const ClusteringType clustering_type) {
 #endif
 }
 
-#if 0
 void
-KnowhereConfig::SetStatisticsLevel(const int64_t stat_level) {
+KnowhereConfig::SetStatisticsLevel(const int32_t stat_level) {
 #ifdef __APPLE__
     // do nothing
 #elif __linux__
@@ -120,7 +119,6 @@ KnowhereConfig::SetStatisticsLevel(const int64_t stat_level) {
     KNOWHERE_THROW_MSG("Unsupported SetStatisticsLevel on current platform!");
 #endif
 }
-#endif
 
 void
 KnowhereConfig::SetLogHandler() {

diff --git a/knowhere/archive/KnowhereConfig.h b/knowhere/archive/KnowhereConfig.h
@@ -62,10 +62,8 @@ class KnowhereConfig {
     /**
      * set Statistics Level [0, 3]
      */
-#if 0
     static void
-    SetStatisticsLevel(const int64_t stat_level);
-#endif
+    SetStatisticsLevel(const int32_t stat_level);
 
     // todo: add log level?
     /**

diff --git a/thirdparty/faiss/faiss/FaissHook.cpp b/thirdparty/faiss/faiss/FaissHook.cpp
@@ -11,6 +11,8 @@
 
 namespace faiss {
 
+int32_t STATISTICS_LEVEL = 0;
+
 /* set default to AVX */
 sq_get_distance_computer_func_ptr sq_get_distance_computer = sq_get_distance_computer_avx;
 sq_sel_quantizer_func_ptr sq_sel_quantizer = sq_select_quantizer_avx;

diff --git a/thirdparty/faiss/faiss/FaissHook.h b/thirdparty/faiss/faiss/FaissHook.h
@@ -11,6 +11,8 @@
 
 namespace faiss {
 
+extern int32_t STATISTICS_LEVEL;
+
 typedef SQDistanceComputer* (*sq_get_distance_computer_func_ptr)(MetricType, QuantizerType, size_t, const std::vector<float>&);
 typedef Quantizer* (*sq_sel_quantizer_func_ptr)(QuantizerType, size_t, const std::vector<float>&);
 typedef InvertedListScanner* (*sq_sel_inv_list_scanner_func_ptr)(MetricType, const ScalarQuantizer*, const Index*, size_t, bool, bool);