mirror of
https://github.com/facebookresearch/faiss.git
synced 2026-10-12 02:13:36 +00:00
Fix avx512 unit test (#5161)
Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/5161 T225351715 fix task as part of oncall duty Reviewed By: mnorris11 Differential Revision: D103035738 fbshipit-source-id: 70b0400be265fe1595ec880ee8cc92fbc177e924
This commit is contained in:
committed by
meta-codesync[bot]
parent
71448c07d3
commit
8380e2543d
@@ -24,7 +24,7 @@ using namespace faiss;
|
||||
// Parameters
|
||||
constexpr int d = 64; // dimension
|
||||
constexpr size_t nb = 1000000; // database size
|
||||
constexpr size_t nt = 10000; // training size
|
||||
constexpr size_t nt = 50000; // training size (>= 39 * nlist for IVF1024)
|
||||
constexpr size_t nq = 100; // number of queries
|
||||
constexpr int nrun = 5; // number of timing runs
|
||||
|
||||
@@ -53,8 +53,7 @@ int main() {
|
||||
|
||||
// Set nprobe for IVF index
|
||||
IndexIVF* index_ivf = dynamic_cast<IndexIVF*>(index.get());
|
||||
if (index_ivf) {
|
||||
}
|
||||
FAISS_THROW_IF_NOT_MSG(index_ivf, "Index is not an IVF index");
|
||||
omp_set_num_threads(1);
|
||||
|
||||
// Test with varying k values
|
||||
|
||||
Reference in New Issue
Block a user