/dports/math/faiss/faiss-1.7.1/faiss/gpu/impl/ |
H A D | IVFPQ.cu | 140 makeTempAlloc(AllocType::Other, stream), in appendVectors_() 147 makeTempAlloc(AllocType::Other, stream), in appendVectors_() 167 makeTempAlloc(AllocType::Other, stream), in appendVectors_() 183 makeTempAlloc(AllocType::Other, stream), in appendVectors_() 187 makeTempAlloc(AllocType::Other, stream), in appendVectors_() 339 makeTempAlloc(AllocType::Quantizer, stream), in setPQCentroids_() 535 makeTempAlloc(AllocType::Other, stream), in query() 539 makeTempAlloc(AllocType::Other, stream), in query() 614 makeTempAlloc(AllocType::Other, stream), in runPQPrecomputedCodes_() 624 makeTempAlloc(AllocType::Other, stream), in runPQPrecomputedCodes_() [all …]
|
H A D | Distance.cu | 73 res, makeTempAlloc(AllocType::Other, stream), {numCentroids}); in runAllPairwiseDistance() 82 res, makeTempAlloc(AllocType::Other, stream), {(int)numQueries}); in runAllPairwiseDistance() 171 res, makeTempAlloc(AllocType::Other, stream), {numCentroids}); in runDistance() 180 res, makeTempAlloc(AllocType::Other, stream), {(int)numQueries}); in runDistance() 208 res, makeTempAlloc(AllocType::Other, stream), {tileRows, tileCols}); in runDistance() 210 res, makeTempAlloc(AllocType::Other, stream), {tileRows, tileCols}); in runDistance() 216 makeTempAlloc(AllocType::Other, stream), in runDistance() 220 makeTempAlloc(AllocType::Other, stream), in runDistance() 227 makeTempAlloc(AllocType::Other, stream), in runDistance() 231 makeTempAlloc(AllocType::Other, stream), in runDistance()
|
H A D | IVFFlatScan.cu | 370 res, makeTempAlloc(AllocType::Other, stream), {kThrustMemSize}); in runIVFFlatScan() 372 res, makeTempAlloc(AllocType::Other, stream), {kThrustMemSize}); in runIVFFlatScan() 412 makeTempAlloc(AllocType::Other, stream), in runIVFFlatScan() 416 makeTempAlloc(AllocType::Other, stream), in runIVFFlatScan() 435 makeTempAlloc(AllocType::Other, stream), in runIVFFlatScan() 439 makeTempAlloc(AllocType::Other, stream), in runIVFFlatScan() 446 makeTempAlloc(AllocType::Other, stream), in runIVFFlatScan() 450 makeTempAlloc(AllocType::Other, stream), in runIVFFlatScan() 457 makeTempAlloc(AllocType::Other, stream), in runIVFFlatScan() 461 makeTempAlloc(AllocType::Other, stream), in runIVFFlatScan()
|
H A D | PQScanMultiPassNoPrecomputed-inl.cuh | 559 res, makeTempAlloc(AllocType::Other, stream), {kThrustMemSize}); in runPQScanMultiPassNoPrecomputed() 603 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() 607 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() 631 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() 640 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() 652 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() 656 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() 663 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() 667 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() 674 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassNoPrecomputed() [all …]
|
H A D | PQScanMultiPassPrecomputed.cu | 578 res, makeTempAlloc(AllocType::Other, stream), {kThrustMemSize}); in runPQScanMultiPassPrecomputed() 580 res, makeTempAlloc(AllocType::Other, stream), {kThrustMemSize}); in runPQScanMultiPassPrecomputed() 620 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassPrecomputed() 624 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassPrecomputed() 643 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassPrecomputed() 647 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassPrecomputed() 654 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassPrecomputed() 658 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassPrecomputed() 665 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassPrecomputed() 669 makeTempAlloc(AllocType::Other, stream), in runPQScanMultiPassPrecomputed()
|
H A D | IVFFlat.cu | 129 makeTempAlloc(AllocType::Other, stream), in appendVectors_() 191 makeTempAlloc(AllocType::Other, stream), in query() 195 makeTempAlloc(AllocType::Other, stream), in query() 211 makeTempAlloc(AllocType::Other, stream), in query()
|
H A D | GeneralDistance.cuh | 317 res, makeTempAlloc(AllocType::Other, stream), {tileRows, tileCols}); in runGeneralDistance() 319 res, makeTempAlloc(AllocType::Other, stream), {tileRows, tileCols}); in runGeneralDistance() 325 makeTempAlloc(AllocType::Other, stream), in runGeneralDistance() 329 makeTempAlloc(AllocType::Other, stream), in runGeneralDistance() 336 makeTempAlloc(AllocType::Other, stream), in runGeneralDistance() 340 makeTempAlloc(AllocType::Other, stream), in runGeneralDistance()
|
H A D | PQCodeDistances-inl.cuh | 414 makeTempAlloc(AllocType::Other, stream), in runPQCodeDistancesMM() 432 makeTempAlloc(AllocType::Other, stream), in runPQCodeDistancesMM() 457 makeTempAlloc(AllocType::Other, stream), in runPQCodeDistancesMM() 478 makeTempAlloc(AllocType::Other, stream), in runPQCodeDistancesMM() 514 makeTempAlloc(AllocType::Other, stream), in runPQCodeDistancesMM() 528 makeTempAlloc(AllocType::Other, stream), in runPQCodeDistancesMM()
|
H A D | IVFBase.cu | 178 makeTempAlloc(AllocType::Other, stream), in updateDeviceListInfo_() 182 makeTempAlloc(AllocType::Other, stream), in updateDeviceListInfo_() 186 makeTempAlloc(AllocType::Other, stream), in updateDeviceListInfo_() 190 makeTempAlloc(AllocType::Other, stream), in updateDeviceListInfo_() 443 makeTempAlloc(AllocType::Other, stream), in addVectors() 448 makeTempAlloc(AllocType::Other, stream), in addVectors()
|
H A D | IVFInterleaved.cuh | 390 makeTempAlloc(AllocType::Other, stream), \ 394 makeTempAlloc(AllocType::Other, stream), \
|
H A D | IVFAppend.cu | 495 makeTempAlloc(AllocType::Other, stream), in runIVFFlatInterleavedAppend() 504 makeTempAlloc(AllocType::Other, stream), in runIVFFlatInterleavedAppend()
|
/dports/math/faiss/faiss-1.7.1/faiss/gpu/utils/ |
H A D | CopyUtils.cuh | 36 resources, makeTempAlloc(AllocType::Other, stream), sizes); in toDeviceTemporary() 81 makeTempAlloc(AllocType::Other, stream), in toDeviceTemporary()
|
H A D | ConversionOperators.cuh | 137 res, makeTempAlloc(AllocType::Other, stream), in.sizes()); in convertTensorTemporary()
|
/dports/math/faiss/faiss-1.7.1/faiss/gpu/ |
H A D | GpuIndex.cu | 220 makeTempAlloc(AllocType::Other, stream), in assign() 388 makeTempAlloc(AllocType::Other, defaultStream), in searchFromCpuPaged_() 392 makeTempAlloc(AllocType::Other, defaultStream), in searchFromCpuPaged_()
|
H A D | GpuResources.h | 122 AllocInfo makeTempAlloc(AllocType at, cudaStream_t st);
|
H A D | GpuResources.cpp | 68 AllocInfo makeTempAlloc(AllocType at, cudaStream_t st) { in makeTempAlloc() function
|
H A D | GpuDistance.cu | 100 makeTempAlloc(AllocType::Other, stream), in bfKnnConvert()
|
H A D | GpuIndexBinaryFlat.cu | 190 makeTempAlloc(AllocType::Other, stream), in search()
|
H A D | GpuIndexFlat.cu | 231 resources_.get(), makeTempAlloc(AllocType::Other, stream), {n, k}); in searchImpl_()
|