Home
last modified time | relevance | path

Searched refs:NUM_THREAD_Q (Results 1 – 5 of 5) sorted by relevance

/dports/math/faiss/faiss-1.7.1/faiss/gpu/impl/
H A DIVFInterleaved.cuh216 #define IVFINT_RUN(CODEC_TYPE, METRIC_TYPE, THREADS, NUM_WARP_Q, NUM_THREAD_Q) \ argument
225 NUM_THREAD_Q, \
243 NUM_THREAD_Q, \
270 #define IVFINT_CODECS(METRIC_TYPE, THREADS, NUM_WARP_Q, NUM_THREAD_Q) \ argument
276 CodecT, METRIC_TYPE, THREADS, NUM_WARP_Q, NUM_THREAD_Q); \
291 NUM_THREAD_Q); \
306 NUM_THREAD_Q); \
317 NUM_THREAD_Q); \
381 #define IVFINT_METRICS(THREADS, NUM_WARP_Q, NUM_THREAD_Q) \ argument
399 IVFINT_CODECS(L2Distance, THREADS, NUM_WARP_Q, NUM_THREAD_Q); \
[all …]
H A DIVFUtilsSelect1.cu103 #define RUN_PASS(BLOCK, NUM_WARP_Q, NUM_THREAD_Q, DIR) \ in runPass1SelectLists() argument
105 pass1SelectLists<BLOCK, NUM_WARP_Q, NUM_THREAD_Q, DIR> \ in runPass1SelectLists()
H A DIVFUtilsSelect2.cu166 #define RUN_PASS(BLOCK, NUM_WARP_Q, NUM_THREAD_Q, DIR) \ in runPass2SelectLists() argument
168 pass2SelectLists<BLOCK, NUM_WARP_Q, NUM_THREAD_Q, DIR> \ in runPass2SelectLists()
H A DIVFInterleaved.cu134 #define IVF_SCAN_2(THREADS, NUM_WARP_Q, NUM_THREAD_Q) \ in runIVFInterleavedScan2() argument
135 ivfInterleavedScan2<THREADS, NUM_WARP_Q, NUM_THREAD_Q> \ in runIVFInterleavedScan2()
H A DL2Select.cu206 #define RUN_L2_SELECT(BLOCK, NUM_WARP_Q, NUM_THREAD_Q) \ in runL2SelectMin() argument
208 l2SelectMinK<T, NUM_WARP_Q, NUM_THREAD_Q, BLOCK> \ in runL2SelectMin()