Holder of GPU resources for a particular flat index. More...
Public Member Functions | |
| FlatIndex (GpuResources *res, int dim, bool l2Distance, bool useFloat16, bool useFloat16Accumulator, bool storeTransposed, MemorySpace space) | |
| bool | getUseFloat16 () const |
| 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< float, 2, true > & | getVectorsFloat32Ref () |
| Returns a reference to our vectors currently in use. | |
| DeviceTensor< float, 2, true > | getVectorsFloat32Copy (cudaStream_t stream) |
| DeviceTensor< float, 2, true > | getVectorsFloat32Copy (int from, int num, cudaStream_t stream) |
| Returns only a subset of the vectors. | |
| void | query (Tensor< float, 2, true > &vecs, int k, Tensor< float, 2, true > &outDistances, Tensor< int, 2, true > &outIndices, bool exactDistance, int tileSize=-1) |
| void | add (const float *data, int numVecs, cudaStream_t stream) |
| void | reset () |
| Free all storage. | |
Holder of GPU resources for a particular flat index.
Definition at line 23 of file FlatIndex.cuh.
| void faiss::gpu::FlatIndex::add | ( | const float * | data, |
| int | numVecs, | ||
| cudaStream_t | stream | ||
| ) |
Add vectors to ourselves; the pointer passed can be on the host or the device
Definition at line 201 of file FlatIndex.cu.
| DeviceTensor< float, 2, true > faiss::gpu::FlatIndex::getVectorsFloat32Copy | ( | cudaStream_t | stream | ) |
Performs a copy of the vectors on the given device, converting as needed from float16
Definition at line 91 of file FlatIndex.cu.
1.8.5