11 #include "../GpuIndicesOptions.h"
12 #include "../utils/Tensor.cuh"
13 #include <thrust/device_vector.h>
15 namespace faiss {
namespace gpu {
18 void runUpdateListPointers(Tensor<int, 1, true>& listIds,
19 Tensor<int, 1, true>& newListLength,
20 Tensor<void*, 1, true>& newCodePointers,
21 Tensor<void*, 1, true>& newIndexPointers,
22 thrust::device_vector<int>& listLengths,
23 thrust::device_vector<void*>& listCodes,
24 thrust::device_vector<void*>& listIndices,
30 void runIVFPQInvertedListAppend(Tensor<int, 1, true>& listIds,
31 Tensor<int, 1, true>& listOffset,
32 Tensor<int, 2, true>& encodings,
33 Tensor<long, 1, true>& indices,
34 thrust::device_vector<void*>& listCodes,
35 thrust::device_vector<void*>& listIndices,
36 IndicesOptions indicesOptions,
40 void runIVFFlatInvertedListAppend(Tensor<int, 1, true>& listIds,
41 Tensor<int, 1, true>& listOffset,
42 Tensor<float, 2, true>& vecs,
43 Tensor<long, 1, true>& indices,
45 thrust::device_vector<void*>& listData,
46 thrust::device_vector<void*>& listIndices,
47 IndicesOptions indicesOptions,