Struct faiss::gpu::IndexWrapper
-
template<typename GpuIndex>
struct faiss::gpu::IndexWrapper Public Functions
-
IndexWrapper(int numGpus, std::function<std::unique_ptr<GpuIndex>(GpuResourcesProvider*, int)> init)
-
void runOnIndices(std::function<void(GpuIndex*)> f)
-
void setNumProbes(int nprobe)
Public Members
-
std::vector<std::unique_ptr<faiss::gpu::StandardGpuResources>> resources
-
std::vector<std::unique_ptr<GpuIndex>> subIndex
-
std::unique_ptr<faiss::IndexReplicas> replicaIndex
-
IndexWrapper(int numGpus, std::function<std::unique_ptr<GpuIndex>(GpuResourcesProvider*, int)> init)