Commit 3888f9b
Using expanded distance forms in
Summary:
This is a minor bug that comes with a perf impact. The classic FAISS `FlatIndex` always uses expanded form of distance computation even though an argument `exactDistances` is provided. `RaftFlatIndex` was using this argument to determine whether the computation should be exhaustive.
This PR includes one additional change to eagerly initialize the `cublas_handle` on the `device_resources` instance when it's created.
Pull Request resolved: facebookresearch#3021
Reviewed By: pemazare
Differential Revision: D48739660
Pulled By: mdouze
fbshipit-source-id: a361334eb243df86c169c69d24bb10fed8876ee9RaftFlatIndex.cu (facebookresearch#3021)1 parent fef49a6 commit 3888f9b
2 files changed
+4
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
432 | 432 | | |
433 | 433 | | |
434 | 434 | | |
| 435 | + | |
| 436 | + | |
| 437 | + | |
435 | 438 | | |
436 | 439 | | |
437 | 440 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
36 | 36 | | |
37 | 37 | | |
38 | 38 | | |
39 | | - | |
40 | | - | |
| 39 | + | |
41 | 40 | | |
42 | 41 | | |
43 | 42 | | |
| |||
0 commit comments