Home
last modified time | relevance | path

Searched refs:makeTempAlloc (Results 1 – 19 of 19) sorted by relevance

/dports/math/faiss/faiss-1.7.1/faiss/gpu/impl/
H A DIVFPQ.cu140 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 DDistance.cu73 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 DIVFFlatScan.cu370 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 DPQScanMultiPassNoPrecomputed-inl.cuh559 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 DPQScanMultiPassPrecomputed.cu578 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 DIVFFlat.cu129 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 DGeneralDistance.cuh317 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 DPQCodeDistances-inl.cuh414 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 DIVFBase.cu178 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 DIVFInterleaved.cuh390 makeTempAlloc(AllocType::Other, stream), \
394 makeTempAlloc(AllocType::Other, stream), \
H A DIVFAppend.cu495 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 DCopyUtils.cuh36 resources, makeTempAlloc(AllocType::Other, stream), sizes); in toDeviceTemporary()
81 makeTempAlloc(AllocType::Other, stream), in toDeviceTemporary()
H A DConversionOperators.cuh137 res, makeTempAlloc(AllocType::Other, stream), in.sizes()); in convertTensorTemporary()
/dports/math/faiss/faiss-1.7.1/faiss/gpu/
H A DGpuIndex.cu220 makeTempAlloc(AllocType::Other, stream), in assign()
388 makeTempAlloc(AllocType::Other, defaultStream), in searchFromCpuPaged_()
392 makeTempAlloc(AllocType::Other, defaultStream), in searchFromCpuPaged_()
H A DGpuResources.h122 AllocInfo makeTempAlloc(AllocType at, cudaStream_t st);
H A DGpuResources.cpp68 AllocInfo makeTempAlloc(AllocType at, cudaStream_t st) { in makeTempAlloc() function
H A DGpuDistance.cu100 makeTempAlloc(AllocType::Other, stream), in bfKnnConvert()
H A DGpuIndexBinaryFlat.cu190 makeTempAlloc(AllocType::Other, stream), in search()
H A DGpuIndexFlat.cu231 resources_.get(), makeTempAlloc(AllocType::Other, stream), {n, k}); in searchImpl_()