use same matching policy in batch

worldcoin · Aug 19, 2024 · f6b00bd · f6b00bd
1 parent 5c46395
commit f6b00bd
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/iris-mpc-gpu/src/dot/kernel.cu b/iris-mpc-gpu/src/dot/kernel.cu
@@ -66,7 +66,7 @@ extern "C" __global__ void mergeDbResults(unsigned long long *matchResultsLeft,
             if (queryIdx >= queryLength || dbIdx >= dbLength)
                 continue;
 
-            // Current *AND* policy: only match, if both eyes match
+            // Current *AND* policy: only match if both eyes match
             if (matchLeft && matchRight)
                 atomicMin(&finalResults[queryIdx], dbIdx);
         }
@@ -98,10 +98,9 @@ extern "C" __global__ void mergeBatchResults(unsigned long long *matchResultsSel
             bool matchLeft = (matchResultsSelfLeft[idx] & (1ULL << i));
             bool matchRight = (matchResultsSelfRight[idx] & (1ULL << i));
 
-            if (matchLeft || matchRight)
-            {
+            // Current *AND* policy: only match if both eyes match
+            if (matchLeft && matchRight)
                 atomicMin(&finalResults[queryIdx], UINT_MAX - 1);
-            }
         }
     }
 }