Home
last modified time | relevance | path

Searched refs:faiss (Results 1 – 25 of 485) sorted by relevance

12345678910>>...20

/dports/math/faiss/faiss-1.7.1/tests/
H A Dtest_factory.py9 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 Dtest_build_blocks.py10 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 Dtest_index.py12 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 Dtest_index_accuracy.py12 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 Dtest_io.py8 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 Dtest_clustering.py10 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 Dtest_autotune.py8 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 Dtest_index_composite.py11 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 Dtest_referenced_objects.py11 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 Dtest_fast_scan.py11 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 Dtest_extra_distances.py10 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 Dtest_index_binary_from_float.py10 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 Dtest_index_binary.py12 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 Dtest_merge.cpp51 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 Dswigfaiss.swig447 %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 DTestGpuIndexIVFFlat.cpp42 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 Dtest_gpu_basics.py10 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 DTestGpuIndexIVFPQ.cpp68 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 DTestGpuIndexFlat.cpp32 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 Dtest_gpu_index.py12 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 DTestGpuIndexIVFScalarQuantizer.cpp24 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 DTestGpuIndexBinaryFlat.cpp31 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 Dcombined_index.py9 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 Dbench_all_ivf.py11 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 DDistance.cuh14 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 …]

12345678910>>...20