fix mac CI

OpenMP on Mac doesn't accept the expression includes a cast as loop condition with pragma omp parallel for
facebookresearch · May 31, 2023 · 632b3bb · 632b3bb
1 parent b48a671
commit 632b3bb
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/faiss/utils/utils.cpp b/faiss/utils/utils.cpp
@@ -447,10 +447,12 @@ uint64_t bvec_checksum(size_t n, const uint8_t* a) {
 }
 
 void bvecs_checksum(size_t n, size_t d, const uint8_t* a, uint64_t* cs) {
-#pragma omp parallel for if (n > 1000)
-    for (std::make_signed<std::size_t>::type i_ = 0;
-         static_cast<std::size_t>(i_) < n;
-         i_++) {
+    // MSVC can't accept unsigned index for #pragma omp parallel for
+    // so below codes only accepts n <= std::numeric_limits<ssize_t>::max()
+    using ssize_t = std::make_signed<std::size_t>::type;
+    const ssize_t size = n;
+#pragma omp parallel for if (size > 1000)
+    for (ssize_t i_ = 0; i_ < size; i_++) {
         const auto i = static_cast<std::size_t>(i_);
         cs[i] = bvec_checksum(d, a + i * d);
     }