mirror of
https://github.com/facebookresearch/faiss.git
synced 2026-10-12 02:13:36 +00:00
Parameterize existing tests with @for_all_simd_levels (#5134)
Summary: Pull Request resolved: https://github.com/facebookresearch/faiss/pull/5134 Expand DD test coverage by applying for_all_simd_levels to existing test classes that exercise DD-dispatched code paths but were previously pinned to a single SIMD level. This is the "mega decorator" diff from the DD test coverage gaps plan -- pure decorator additions, no new test logic. Follow-up diffs add new test files and numerical cross-level assertions for gaps the decorator alone cannot close. Classes decorated (grouped by area): * Binary Hamming non-IVF: TestRange and TestKnn in test_binary_hashindex.py; TestBinarySearchParams in test_binary_search_params.py; TestIndexBinaryFromFloat in test_index_binary_from_float.py; TestSpectralHash in test_index_accuracy.py. * IVFPQ / search: EvalIVFPQAccuracy in test_index.py; TestSelector and TestSearchParams in test_search_params.py. * Flat / refine / Panorama: TestIndexFlat, TestIndexFlatL2, TestIndexFlatL2Panorama, TestScalarQuantizer in test_index.py; TestDistanceComputer, TestIndexRefineSearchParams, TestIndexRefineRangeSearch in test_refine.py; TestIndexRefinePanorama, TestIndexFlatPanorama, TestIndexHNSWFlatPanorama, TestIndexIVFFlatPanorama in their respective files. * Quantizer encode: TestResidualQuantizer, TestIndexResidualQuantizerSearch in test_residual_quantizer.py; TestComponents, TestLocalSearchQuantizer in test_local_search_quantizer.py. * Fast scan: TestFastScanFiltering, TestBlockSkipConsistency, TestFastScanRangeSearchFilter in test_fastscan_filter.py. * Broader index tests (search-exercising classes only): TestParameterSpace in test_autotune.py; TestSpectralHash in test_factory.py; TestMerge1, TestMerge2, TestRemoveFastScan in test_merge_index.py. BUCK changes move decorated tests from the legacy-listing lists to the simd_levels lists (which use supports_static_listing = False, required for dynamic class name generation), and add the new entries test_binary_search_params, test_fastscan_filter, test_refine_panorama, test_hnsw_panorama. ___ overriding_review_checks_triggers_an_audit_and_retroactive_review Oncall Short Name: fair_umami_cluster Differential Revision: D101822335 fbshipit-source-id: 9a432c6d6bee201c3731713226405a8c8ecebbe6
This commit is contained in:
committed by
meta-codesync[bot]
parent
8312540763
commit
f4338ade1b
@@ -13,6 +13,7 @@ import platform
|
||||
import faiss
|
||||
import unittest
|
||||
|
||||
from common_faiss_tests import for_all_simd_levels
|
||||
from faiss.contrib import datasets
|
||||
|
||||
faiss.omp_set_num_threads(4)
|
||||
@@ -117,6 +118,7 @@ def icm_encode_ref(x, codebooks, codes):
|
||||
return codes
|
||||
|
||||
|
||||
@for_all_simd_levels
|
||||
class TestComponents(unittest.TestCase):
|
||||
|
||||
def test_decode(self):
|
||||
@@ -332,6 +334,7 @@ def eval_codec(q, xb):
|
||||
return ((xb - decoded) ** 2).sum()
|
||||
|
||||
|
||||
@for_all_simd_levels
|
||||
class TestLocalSearchQuantizer(unittest.TestCase):
|
||||
|
||||
def test_training(self):
|
||||
|
||||
Reference in New Issue
Block a user