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