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 D | IVFInterleaved.cuh | 216 #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 D | IVFUtilsSelect1.cu | 103 #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 D | IVFUtilsSelect2.cu | 166 #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 D | IVFInterleaved.cu | 134 #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 D | L2Select.cu | 206 #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()
|