1 files changed, 6 insertions, 8 deletions
diff --git a/eval/src/tests/ann/sift_benchmark.cpp b/eval/src/tests/ann/sift_benchmark.cpp
index 7c060d86371..f3570eb9247 100644
--- a/eval/src/tests/ann/sift_benchmark.cpp
+++ b/eval/src/tests/ann/sift_benchmark.cpp
@@ -8,7 +8,6 @@
 #include <unistd.h>
 #include <stdio.h>
 #include <chrono>
-#include <cstdlib>
 
 #define NUM_DIMS 128
 #define NUM_DOCS 1000000
@@ -29,12 +28,11 @@ struct PointVector {
 };
 
 static PointVector *aligned_alloc(size_t num) {
-    size_t sz = num * sizeof(PointVector);
-    size_t align = 512;
-    while ((sz % align) != 0) { align /= 2; }
-    double mega_bytes = sz / (1024.0*1024.0);
-    fprintf(stderr, "allocate %.2f MB of vectors (align %zu)\n", mega_bytes, align);
-    void *mem = std::aligned_alloc(align, sz);
+    char *mem = (char *)malloc(num * sizeof(PointVector) + 512);
+    mem += 512;
+    size_t val = (size_t)mem;
+    size_t unalign = val % 512;
+    mem -= unalign;
     return reinterpret_cast<PointVector *>(mem);
 }
 
@@ -171,7 +169,7 @@ void verifyBF(uint32_t qid) {
             fprintf(stderr, "WARN dist %.9g < mindist %.9g\n", dist, min_distance);
         }
         EXPECT_FALSE(dist+0.000001 < min_distance);
-        if (min_distance > 0) all_c2.push_back(dist / min_distance);
+        if (qid == 6) all_c2.push_back(dist / min_distance);
     }
     if (all_c2.size() != NUM_DOCS) return;
     std::sort(all_c2.begin(), all_c2.end());