faiss/tests
Tarang Jain 27b1055cc6 Integrate IVF-PQ from RAFT (#3044)
Summary:
Imports changes from https://github.com/facebookresearch/faiss/issues/3133 and https://github.com/facebookresearch/faiss/issues/3171. So this single PR adds all the changes together.

- [x] Implement RaftIVFPQ class
- [x] Update gtests to test correctness with RAFT enabled
- [x] All googleTests for RAFT enabled IVFPQ pass
- [x] Move some common functions in RaftIVFFlat and RaftIVFPQ to helper: RaftUtils.h
- [x] update Quantizer retroactively after building RAFT index -- both IVFFlat and IVFPQ
- [x] resolve failing LargeBatch (classical GPU)
- [x] add checks for Pascal deprecation
- [x] apply RMM changes from https://github.com/facebookresearch/faiss/issues/3171
- [x] apply robertmaynard's changes from https://github.com/facebookresearch/faiss/issues/3133

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

Reviewed By: junjieqi

Differential Revision: D51074065

Pulled By: algoriddle

fbshipit-source-id: 6871257921bcaff2064a20637e2ed358acbdc363
2024-02-21 06:41:08 -08:00
..
CMakeLists.txt Integrate IVF-PQ from RAFT (#3044) 2024-02-21 06:41:08 -08:00
common_faiss_tests.py
test_RCQ_cropping.cpp
test_approx_topk.cpp Remove unused exception parameter from files inc faiss/tests/test_approx_topk.cpp 2023-12-08 05:37:19 -08:00
test_autotune.py
test_binary_factory.py
test_binary_flat.cpp
test_binary_hashindex.py
test_binary_io.py
test_build_blocks.py
test_clone.py
test_clustering.py
test_code_distance.cpp
test_contrib.py Index pretransform support in search_preassigned (#3225) 2024-01-30 09:20:07 -08:00
test_contrib_with_scipy.py
test_cppcontrib_sa_decode.cpp
test_cppcontrib_uintreader.cpp
test_dealloc_invlists.cpp
test_distances_simd.cpp
test_documentation.py
test_doxygen_documentation.py
test_extra_distances.py
test_factory.py
test_fast_scan.py
test_fast_scan_ivf.py Generalize ResultHanlder, support range search for HNSW and Fast Scan (#3190) 2024-01-11 11:46:30 -08:00
test_fastscan_perf.cpp fix perf test (#3255) 2024-02-15 10:38:55 -08:00
test_graph_based.py Generalize ResultHanlder, support range search for HNSW and Fast Scan (#3190) 2024-01-11 11:46:30 -08:00
test_heap.cpp
test_hnsw.cpp
test_index.py Generalize ResultHanlder, support range search for HNSW and Fast Scan (#3190) 2024-01-11 11:46:30 -08:00
test_index_accuracy.py avx512 CI + conda packages (#3197) 2024-01-11 08:26:33 -08:00
test_index_binary.py
test_index_binary_from_float.py
test_index_composite.py Search and return codes (#3143) 2023-11-25 13:57:25 -08:00
test_io.py
test_ivf_index.cpp add a context parameter to InvertedLists and InvertedListsIterator (#3247) 2024-02-09 09:14:38 -08:00
test_ivflib.py
test_ivfpq_codec.cpp
test_ivfpq_indexing.cpp
test_local_search_quantizer.py
test_lowlevel_ivf.cpp
test_mem_leak.cpp
test_merge.cpp
test_merge_index.py
test_meta_index.py
test_omp_threads.cpp
test_omp_threads_py.py
test_ondisk_ivf.cpp Remove semicolon(s) from 3 files inc facer/engine/utils/UDPServer.h 2024-02-14 14:37:40 -08:00
test_oom_exception.py
test_pairs_decoding.cpp
test_params_override.cpp
test_partition.py
test_partitioning.cpp
test_pq_encoding.cpp
test_product_quantizer.py
test_referenced_objects.py
test_refine.py Add search parameters for IndexRefine::search() and IndexRefineFlat::search() (#3122) 2023-11-05 15:07:39 -08:00
test_residual_quantizer.py fix test TestCrossCodebookComputations::test_precomp 2023-10-31 09:50:05 -07:00
test_rowwise_minmax.py
test_search_params.py fix ACCESS VIOLATION error when searching using IDSelectorArray 2024-01-29 23:08:45 -08:00
test_simdlib.cpp
test_sliding_ivf.cpp
test_standalone_codec.py Search and return codes (#3143) 2023-11-25 13:57:25 -08:00
test_threaded_index.cpp Remove unused variables in faiss/tests/test_threaded_index.cpp 2024-02-14 15:59:23 -08:00
test_transfer_invlists.cpp
torch_test_contrib.py