/dports/math/faiss/faiss-1.7.1/tests/ |
H A D | test_factory.py | 9 import faiss 11 from faiss.contrib import factory_tools 24 index = faiss.index_factory(12, "PQ4") 39 index = faiss.index_factory(12, "SQ8") 67 index = faiss.index_factory(12, "HNSW32,SQ8", faiss.METRIC_INNER_PRODUCT) 81 index = faiss.index_factory(12, "NSG64", faiss.METRIC_INNER_PRODUCT) 109 index = faiss.index_factory(56, "IVF50,PQ28x4fs_64", faiss.METRIC_INNER_PRODUCT) 165 xb = faiss.rand((100, 20)) 168 index2 = faiss.clone_index(index) 175 xb = faiss.rand((1000, 16)) [all …]
|
H A D | test_build_blocks.py | 10 import faiss 26 pca = faiss.PCAMatrix(d, 10) 44 index = faiss.IndexFlatL2(4) 83 m = faiss.MapLong2Long() 140 swig_ptr = faiss.swig_ptr 225 faiss.normalize_L2(m) 302 x = faiss.rand(2000) 362 hnsw.neighbor_range(23, 0, faiss.swig_ptr(be), faiss.swig_ptr(be[1:])) 381 faiss.IndexScalarQuantizer(10, faiss.ScalarQuantizer.QT_8bit) 394 ptr = faiss.swig_ptr(a) [all …]
|
H A D | test_index.py | 12 import faiss 336 index = faiss.IndexScalarQuantizer(d, qtype, faiss.METRIC_L2) 361 index = faiss.IndexFlatL2(d) 425 index = faiss.IndexFlatL2(d) 439 index = faiss.IndexIVFFlat(quantizer, d, 32, faiss.METRIC_L2) 505 index = faiss.IndexFlatL2(d) 583 index = faiss.IndexHNSWFlat(d, 16, faiss.METRIC_INNER_PRODUCT) 650 vt = faiss.VisitedTable(nb) 736 metric = faiss.METRIC_L2 913 sel = faiss.IDSelectorArray(50, faiss.swig_ptr(toremove[:50])) [all …]
|
H A D | test_index_accuracy.py | 12 import faiss 41 q = faiss.IndexFlatL2(d) 47 ivfk = faiss.IndexIVFFlat(faiss.IndexFlatL2(d), d, ncentroids) 63 q = faiss.IndexLSH(d, nbits) 87 q = faiss.IndexPQ(d, M, nbits_per_index, faiss.METRIC_INNER_PRODUCT) 95 ivfpq = faiss.IndexIVFPQ(faiss.IndexFlatL2(d), d, ncentroids, M, 8) 111 rq = faiss.IndexRefineFlat(q) 379 for metric_type in faiss.METRIC_L2, faiss.METRIC_INNER_PRODUCT: 572 ref = faiss.IndexFlatL2(1) 576 new = faiss.IndexFlat1D() [all …]
|
H A D | test_io.py | 8 import faiss 24 index = faiss.IndexFlatL2(d) 29 faiss.write_index(index, fname) 32 faiss.read_index(fname) 42 faiss.read_index(fname) 59 index = faiss.IndexFlatL2(d) 69 faiss.write_index(index, writer) 81 faiss.vector_to_array(index.xb) == faiss.vector_to_array(index2.xb) 117 index = faiss.IndexFlatL2(d) 166 index = faiss.IndexFlatL2(d) [all …]
|
H A D | test_clustering.py | 10 import faiss 84 faiss.normalize_L2(ccent) 89 clus = faiss.Clustering(d, 5) 90 index = faiss.IndexFlatL2(d) 104 clus = faiss.Clustering(d, 5) 105 index = faiss.IndexFlatL2(d) 129 codec = faiss.IndexScalarQuantizer(d, faiss.ScalarQuantizer.QT_4bit) 138 index = faiss.IndexFlatL2(d) 148 index = faiss.IndexFlatL2(d) 209 faiss.normalize_L2(x) [all …]
|
H A D | test_autotune.py | 8 import faiss 14 index = faiss.index_factory(32, "IVF32,Flat") 15 ps = faiss.ParameterSpace() 21 ps = faiss.ParameterSpace() 23 index2 = faiss.downcast_index(index.index) 28 ps = faiss.ParameterSpace() 30 index2 = faiss.downcast_index(index.base_index) 35 ps = faiss.ParameterSpace() 39 index2 = faiss.downcast_index(index2.index) 44 ps = faiss.ParameterSpace() [all …]
|
H A D | test_index_composite.py | 11 import faiss 29 quantizer = faiss.IndexFlatL2(d) 75 index = faiss.IndexFlat(5) 84 sub_index = faiss.IndexFlat(5) 302 faiss.normalize_L2(x) 304 faiss.normalize_L2(x) 307 index2 = faiss.IndexFlatL2(2) 354 index3 = faiss.read_index(fname, faiss.IO_FLAG_MMAP) 458 index = faiss.IndexFlatL2(d) 463 writer = faiss.VectorIOWriter() [all …]
|
H A D | test_referenced_objects.py | 11 import faiss 23 quantizer = faiss.IndexFlatL2(d) 32 quantizer = faiss.IndexFlatL2(d) 41 index = faiss.IndexIVFFlat(faiss.IndexFlatL2(d), d, 10) 47 sub_index = faiss.IndexFlatL2(d) 58 sub_index = faiss.IndexFlatL2(d) 71 sub_index = faiss.IndexFlatL2(d) 72 index = faiss.IndexIDMap(sub_index) 79 index = faiss.IndexShards(d) 96 index = faiss.IndexBinaryIVF(faiss.IndexBinaryFlat(dbin), dbin, 10) [all …]
|
H A D | test_fast_scan.py | 11 import faiss 13 from faiss.contrib import datasets 20 options = faiss.get_compile_options() 31 index_gt = faiss.IndexFlatL2(32) 95 index2 = faiss.IndexPQFastScan(index) 177 sp = faiss.swig_ptr 180 faiss.pq4_pack_LUT( 186 faiss.pq4_pack_codes( 275 index2 = faiss.IndexPQFastScan(index) 284 index2 = faiss.IndexPQFastScan(index) [all …]
|
H A D | test_extra_distances.py | 10 import faiss 38 faiss.METRIC_L1) 42 faiss.METRIC_Linf) 50 new_dis = faiss.pairwise_distances(xq, yb, faiss.METRIC_L2) 67 new_dis = faiss.pairwise_distances(xq, yb, faiss.METRIC_Lp, p) 96 index = faiss.IndexFlat(d, mt) 108 index2 = faiss.deserialize_index(faiss.serialize_index(index)) 115 self.do_test_knn(faiss.METRIC_L1) 118 self.do_test_knn(faiss.METRIC_Linf) 132 mt = faiss.METRIC_L1 [all …]
|
H A D | test_index_binary_from_float.py | 10 import faiss 37 index_ref = faiss.IndexFlatL2(d) 40 index = faiss.IndexFlatL2(d) 41 index_bin = faiss.IndexBinaryFromFloat(index) 57 quantizer_ref = faiss.IndexBinaryFlat(d) 63 unwrapped_quantizer = faiss.IndexFlatL2(d) 83 index_ref = faiss.IndexBinaryFlat(d) 125 index_ref = faiss.IndexBinaryFlat(d) 130 clus = faiss.Clustering(d, nlist) 131 clus_index = faiss.IndexFlatL2(d) [all …]
|
H A D | test_index_binary.py | 12 import faiss 63 quantizer = faiss.IndexFlatL2(d) 100 index = faiss.IndexBinaryFlat(d) 115 index = faiss.IndexBinaryFlat(d) 127 index = faiss.IndexBinaryFlat(d) 158 index = faiss.IndexBinaryFlat(d) 166 quantizer = faiss.IndexBinaryFlat(d) 217 index = faiss.IndexBinaryIVF(faiss.IndexBinaryFlat(d), d, 8) 322 faiss.omp_set_num_threads(1) 358 faiss.omp_set_num_threads(1) [all …]
|
H A D | test_merge.cpp | 51 typedef faiss::Index::idx_t idx_t; 65 faiss::IndexFlatL2 quantizer; 91 faiss::IndexShards* index_shards, in compare_merged() 105 faiss::ivflib::merge_into( in compare_merged() 112 faiss::IndexIVF* index0 = nullptr; in compare_merged() 125 auto il = new faiss::OnDiskInvertedLists( in compare_merged() 150 faiss::IndexShards index_shards(d); in TEST() 186 faiss::RandomRotationMatrix rot(d, d); in TEST() 188 faiss::IndexFlatL2 quantizer(d); in TEST() 192 faiss::IndexPreTransform ipt(&rot, &iflat); in TEST() [all …]
|
/dports/math/faiss/faiss-1.7.1/faiss/python/ |
H A D | swigfaiss.swig | 447 %template(ThreadedIndexBase) faiss::ThreadedIndex<faiss::Index>; 451 %template(IndexShards) faiss::IndexShardsTemplate<faiss::Index>; 455 %template(IndexReplicas) faiss::IndexReplicasTemplate<faiss::Index>; 459 %template(IndexIDMap) faiss::IndexIDMapTemplate<faiss::Index>; 461 %template(IndexIDMap2) faiss::IndexIDMap2Template<faiss::Index>; 645 faiss::Index * downcast_index (faiss::Index *index) 653 faiss::IndexBinary * downcast_IndexBinary (faiss::IndexBinary *index) 862 faiss::partition_fuzzy<faiss::CMin<float, int64_t> >; 864 faiss::partition_fuzzy<faiss::CMax<float, int64_t> >; 896 return faiss::partition_fuzzy<faiss::CMax<unsigned short, int> >( [all …]
|
/dports/math/faiss/faiss-1.7.1/faiss/gpu/test/ |
H A D | TestGpuIndexIVFFlat.cpp | 42 device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in Options() 80 faiss::Index* quantizer = metricType == faiss::METRIC_L2 in queryTest() 130 faiss::Index* quantizer = metricType == faiss::METRIC_L2 in addTest() 190 faiss::IndexIVFFlat cpuIndex(&cpuQuantizer, 1, 1, faiss::METRIC_L2); in copyToTest() 208 faiss::gpu::compareIndices( in copyToTest() 241 faiss::gpu::GpuIndexIVFFlat gpuIndex(&res, 1, 1, faiss::METRIC_L2, config); in copyFromTest() 258 faiss::gpu::compareIndices( in copyFromTest() 357 faiss::Index* quantizer = (faiss::Index*)&quantizerIP; in TEST() 386 faiss::gpu::compareIndices( in TEST() 495 int device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in TEST() [all …]
|
H A D | test_gpu_basics.py | 10 import faiss 24 index = faiss.IndexReplicas() 34 index = faiss.index_cpu_to_gpu(faiss.StandardGpuResources(), 0, 39 index = faiss.GpuIndexFlat(faiss.StandardGpuResources(), 69 index = faiss.GpuIndexBinaryFlat(faiss.StandardGpuResources(), 98 sub_index = faiss.GpuIndexFlatIP(faiss.StandardGpuResources(), dim, config) 116 km1 = faiss.Kmeans(d, k) 199 gpu_quantizer = faiss.index_cpu_to_all_gpus(faiss.IndexFlatL2(dim)) 298 faiss.METRIC_L2, 300 faiss.METRIC_L1, [all …]
|
H A D | TestGpuIndexIVFPQ.cpp | 68 device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in Options() 318 faiss::IndexIVFPQ cpuIndex( in TEST() 343 faiss::gpu::compareIndices( in TEST() 519 faiss::IndexIVFPQ cpuIndex( in TEST() 561 faiss::gpu::compareIndices( in TEST() 594 faiss::METRIC_L2, in TEST() 640 faiss::METRIC_L2, in TEST() 677 int device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in TEST() 701 faiss::IndexIVFPQ cpuIndex( in TEST() 721 faiss::METRIC_L2, in TEST() [all …]
|
H A D | TestGpuIndexFlat.cpp | 32 faiss::MetricType metric; 57 faiss::gpu::randVal(1, faiss::gpu::getMaxKSelection()), in testFlat() 68 int device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in testFlat() 93 faiss::gpu::compareIndices( in testFlat() 270 faiss::IndexFlatL2 cpuIndex(dim); in TEST() 279 int device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in TEST() 313 int device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in TEST() 326 faiss::IndexFlatL2 cpuIndex(2000); in TEST() 349 int device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in TEST() 381 faiss::gpu::compareIndices( in TEST() [all …]
|
H A D | test_gpu_index.py | 12 import faiss 44 mat = faiss.PCAMatrix(20, 10) 96 if faiss.get_num_gpus() == 1: 194 … idx_gpu = faiss.GpuIndexIVFPQ(res, d, nlist, sub_q, bits_per_code, faiss.METRIC_L2, config) 195 q = faiss.IndexFlatL2(d) 196 idx_cpu = faiss.IndexIVFPQ(q, d, nlist, sub_q, bits_per_code, faiss.METRIC_L2) 240 q = faiss.IndexFlatL2(d) 241 idx_cpu = faiss.IndexIVFPQ(q, d, nlist, sub_q, bits_per_code, faiss.METRIC_L2) 285 idx_cpu = faiss.IndexIVFPQ(q, d, nlist, sub_q, bits_per_code, faiss.METRIC_L2) 325 idx = faiss.GpuIndexIVFFlat(res, d, nlist, faiss.METRIC_L2, config) [all …]
|
H A D | TestGpuIndexIVFScalarQuantizer.cpp | 24 dim = faiss::gpu::randVal(64, 200); in Options() 35 indicesOpt = faiss::gpu::randSelect( in Options() 36 {faiss::gpu::INDICES_CPU, in Options() 37 faiss::gpu::INDICES_32_BIT, in Options() 40 device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in Options() 61 faiss::gpu::IndicesOptions indicesOpt; 65 using namespace faiss; in runCopyToTest() 66 using namespace faiss::gpu; in runCopyToTest() 145 using namespace faiss; in runCopyFromTest() 146 using namespace faiss::gpu; in runCopyFromTest() [all …]
|
H A D | TestGpuIndexBinaryFlat.cpp | 31 std::set<faiss::IndexBinary::idx_t> cpuLabelSet; in compareBinaryDist() 32 std::set<faiss::IndexBinary::idx_t> gpuLabelSet; in compareBinaryDist() 66 faiss::gpu::StandardGpuResources res; in testGpuIndexBinaryFlat() 69 faiss::gpu::GpuIndexBinaryFlatConfig config; in testGpuIndexBinaryFlat() 70 config.device = faiss::gpu::randVal(0, faiss::gpu::getNumDevices() - 1); in testGpuIndexBinaryFlat() 73 int dims = faiss::gpu::randVal(1, 20) * DimMultiple; in testGpuIndexBinaryFlat() 76 faiss::IndexBinaryFlat cpuIndex(dims); in testGpuIndexBinaryFlat() 80 : faiss::gpu::randVal(1, faiss::gpu::getMaxKSelection()); in testGpuIndexBinaryFlat() 81 int numVecs = faiss::gpu::randVal(k + 1, 20000); in testGpuIndexBinaryFlat() 82 int numQuery = faiss::gpu::randVal(1, 1000); in testGpuIndexBinaryFlat() [all …]
|
/dports/math/faiss/faiss-1.7.1/benchs/distributed_ondisk/ |
H A D | combined_index.py | 9 import faiss 25 ilv = faiss.InvertedListsPtrVector() 30 index = faiss.read_index(fname) 44 faiss.IO_FLAG_MMAP | faiss.IO_FLAG_READ_ONLY) 98 n, faiss.swig_ptr(xq), k, 99 faiss.swig_ptr(list_nos), faiss.swig_ptr(coarse_dis), 100 faiss.swig_ptr(D), faiss.swig_ptr(I), False) 112 res = faiss.RangeSearchResult(n) 115 n, faiss.swig_ptr(xq), radius, 116 faiss.swig_ptr(list_nos), faiss.swig_ptr(coarse_dis), [all …]
|
/dports/math/faiss/faiss-1.7.1/benchs/bench_all_ivf/ |
H A D | bench_all_ivf.py | 11 import faiss 106 if hasattr(faiss, "IndexRefine") and isinstance(index, faiss.IndexRefine): 108 if isinstance(index, faiss.IndexIVF): 117 index = faiss.read_index(args.indexfile) 127 index = faiss.index_factory( 129 faiss.METRIC_INNER_PRODUCT 199 training_pq = faiss.ProductQuantizer( 255 faiss.get_mem_usage_kb())) 265 if isinstance(index_ivf, faiss.IndexIVFPQ): 294 ps = faiss.ParameterSpace() [all …]
|
/dports/math/faiss/faiss-1.7.1/faiss/gpu/impl/ |
H A D | Distance.cuh | 14 namespace faiss { namespace 133 faiss::MetricType metric, in allPairwiseDistanceOnDevice() 142 if ((metric == faiss::MetricType::METRIC_L2) || in allPairwiseDistanceOnDevice() 180 if ((metric == faiss::MetricType::METRIC_L1) || in allPairwiseDistanceOnDevice() 196 } else if (metric == faiss::MetricType::METRIC_Lp) { in allPairwiseDistanceOnDevice() 203 } else if (metric == faiss::MetricType::METRIC_Linf) { in allPairwiseDistanceOnDevice() 253 faiss::MetricType metric, in bfKnnOnDevice() 264 if ((metric == faiss::MetricType::METRIC_L2) || in bfKnnOnDevice() 306 if ((metric == faiss::MetricType::METRIC_L1) || in bfKnnOnDevice() 328 } else if (metric == faiss::MetricType::METRIC_Lp) { in bfKnnOnDevice() [all …]
|