Searched refs:index_gpu (Results 1 – 2 of 2) sorted by relevance
| /dports/math/faiss/faiss-1.7.1/contrib/ |
| H A D | exhaustive_search.py | 52 def range_search_gpu(xq, r2, index_gpu, index_cpu): argument
61 k = min(index_gpu.ntotal, 1024)
62 D, I = index_gpu.search(xq, k)
63 if index_gpu.metric_type == faiss.METRIC_L2:
72 index_cpu = faiss.IndexFlat(d, index_gpu.metric_type)
80 if index_gpu.metric_type == faiss.METRIC_L2:
112 index_gpu = faiss.index_cpu_to_all_gpus(index, co=co, ngpu=ngpu)
121 index_gpu.add(xbi)
122 lims_i, Di, Ii = range_search_gpu(xq, threshold, index_gpu, xbi)
123 index_gpu.reset()
[all …]
|
| /dports/math/faiss/faiss-1.7.1/faiss/python/ |
| H A D | __init__.py | 856 index_gpu = index_cpu_to_gpus_list(index, co=co, gpus=None, ngpu=ngpu)
857 return index_gpu
868 index_gpu = index_cpu_to_gpu_multiple_py(res, index, co, gpus)
869 return index_gpu
|