Fix openMP index bug (#4348)

LKLLLLLLLLLL · facebook-github-bot · commit 7ebd133462fb · 2025-05-21T09:27:12.000-07:00
Summary: Fix issue #4340 Change openMP index from size_t to int, to satisfy openMP requirement. Pull Request resolved: #4348 Reviewed By: mnorris11 Differential Revision: D74885943 Pulled By: ramilbakhshyiev fbshipit-source-id: 4b4999b8e7a1741f5d75bcc4f39c4231be4d0978
diff --git a/faiss/utils/distances_fused/simdlib_based.cpp b/faiss/utils/distances_fused/simdlib_based.cpp
@@ -260,7 +260,7 @@ void exhaustive_L2sqr_fused_cmax(
     const size_t nx_p = (nx / NX_POINTS_PER_LOOP) * NX_POINTS_PER_LOOP;
     // the main loop.
 #pragma omp parallel for schedule(dynamic)
-    for (size_t i = 0; i < nx_p; i += NX_POINTS_PER_LOOP) {
+    for (int64_t i = 0; i < nx_p; i += NX_POINTS_PER_LOOP) {
         kernel<DIM, NX_POINTS_PER_LOOP, NY_POINTS_PER_LOOP>(
                 x, y, y_transposed.data(), ny, res, y_norms, i);
     }

Original file line number	Diff line number	Diff line change
`@@ -260,7 +260,7 @@ void exhaustive_L2sqr_fused_cmax(`
`260`	`260`	`const size_t nx_p = (nx / NX_POINTS_PER_LOOP) * NX_POINTS_PER_LOOP;`
`261`	`261`	`// the main loop.`
`262`	`262`	`#pragma omp parallel for schedule(dynamic)`
`263`		`- for (size_t i = 0; i < nx_p; i += NX_POINTS_PER_LOOP) {`
	`263`	`+ for (int64_t i = 0; i < nx_p; i += NX_POINTS_PER_LOOP) {`
`264`	`264`	`kernel<DIM, NX_POINTS_PER_LOOP, NY_POINTS_PER_LOOP>(`
`265`	`265`	`x, y, y_transposed.data(), ny, res, y_norms, i);`
`266`	`266`	`}`