Missed printing 'D' (#3433)

Summary:
'I' was printed twice and 'D' (distance vector) was not printed. Fixed.

Pull Request resolved: https://github.com/facebookresearch/faiss/pull/3433

Reviewed By: fxdawnn

Differential Revision: D57451544

Pulled By: junjieqi

fbshipit-source-id: fc17b3b467f8b2c4ad7d80b44866456d9146e530
This commit is contained in:
Saarth Deshpande
2024-05-21 13:02:25 -07:00
committed by Facebook GitHub Bot
parent 4489773fd4
commit 59e3ee1e30
3 changed files with 6 additions and 11 deletions
+2 -2
View File
@@ -83,10 +83,10 @@ int main() {
printf("\n");
}
printf("I (5 last results)=\n");
printf("D (5 last results)=\n");
for (int i = nq - 5; i < nq; i++) {
for (int j = 0; j < k; j++)
printf("%5zd ", I[i * k + j]);
printf("%5f ", D[i * k + j]);
printf("\n");
}
+2 -5
View File
@@ -61,13 +61,10 @@ int main() {
printf("\n");
}
index.nprobe = 10;
index.search(nq, xq, k, D, I);
printf("I=\n");
printf("D=\n");
for (int i = nq - 5; i < nq; i++) {
for (int j = 0; j < k; j++)
printf("%5zd ", I[i * k + j]);
printf("%5f ", D[i * k + j]);
printf("\n");
}
+2 -4
View File
@@ -55,12 +55,10 @@ int main() {
printf("\n");
}
index.search(nq, xq, k, D, I);
printf("I=\n");
printf("D=\n");
for (int i = nq - 5; i < nq; i++) {
for (int j = 0; j < k; j++)
printf("%5zd ", I[i * k + j]);
printf("%5f ", D[i * k + j]);
printf("\n");
}