Populate the mapping only if MAP_POPULATE is defined (#68)

Signed-off-by: yah01 <yah2er0ne@outlook.com>
zilliztech · Sep 11, 2023 · 5f337f9 · 5f337f9
1 parent 7667353
commit 5f337f9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/thirdparty/hnswlib/hnswlib/hnswalg.h b/thirdparty/hnswlib/hnswlib/hnswalg.h
@@ -659,7 +659,11 @@ class HierarchicalNSW : public AlgorithmInterface<dist_t> {
 
         auto input = knowhere::FileReader(location);
         map_size_ = input.size();
-        map_ = static_cast<char*>(mmap(nullptr, map_size_, PROT_READ, MAP_SHARED | MAP_POPULATE, input.descriptor(), 0));
+        int map_flags = MAP_SHARED;
+#ifdef MAP_POPULATE
+        map_flags |= MAP_POPULATE;
+#endif
+        map_ = static_cast<char*>(mmap(nullptr, map_size_, PROT_READ, map_flags, input.descriptor(), 0));
         madvise(map_, map_size_, MADV_RANDOM);
 
         size_t dim;