Holder of GPU resources for a particular flat index. More...
Public Member Functions | |
| BinaryFlatIndex (GpuResources *res, int dim, MemorySpace space) | |
| int | getSize () const |
| Returns the number of vectors we contain. | |
| int | getDim () const |
| void | reserve (size_t numVecs, cudaStream_t stream) |
| Reserve storage that can contain at least this many vectors. | |
| Tensor< unsigned char, 2, true > & | getVectorsRef () |
| Returns a reference to our vectors currently in use. | |
| void | query (Tensor< unsigned char, 2, true > &vecs, int k, Tensor< int, 2, true > &outDistances, Tensor< int, 2, true > &outIndices) |
| void | add (const unsigned char *data, int numVecs, cudaStream_t stream) |
| void | reset () |
| Free all storage. | |
Holder of GPU resources for a particular flat index.
Definition at line 21 of file BinaryFlatIndex.cuh.
| void faiss::gpu::BinaryFlatIndex::add | ( | const unsigned char * | data, |
| int | numVecs, | ||
| cudaStream_t | stream | ||
| ) |
Add vectors to ourselves; the pointer passed can be on the host or the device
Definition at line 63 of file BinaryFlatIndex.cu.
1.8.5