Lines Matching refs:index_gpu
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()
215 index_gpu = faiss.index_cpu_to_all_gpus(index, co=co, ngpu=ngpu)
225 lims_i, Di, Ii = range_search_gpu(xqi, radius, index_gpu, index)