Skip to content

Commit

Permalink
Merge branch 'main' into fastscan-reconstruct
Browse files Browse the repository at this point in the history
  • Loading branch information
alisafaya authored Jan 7, 2025
2 parents 74a5893 + 3d0ac32 commit bcc9f32
Showing 1 changed file with 18 additions and 0 deletions.
18 changes: 18 additions & 0 deletions faiss/impl/pq4_fast_scan.h
Original file line number Diff line number Diff line change
Expand Up @@ -189,4 +189,22 @@ void pq4_accumulate_loop_qbs(
SIMDResultHandler& res,
const NormTableScaler* scaler = nullptr);

/** Wrapper of pq4_accumulate_loop_qbs using simple StoreResultHandler
* and DummyScaler
*
* @param nq number of queries
* @param ntotal2 number of database elements (multiple of 32)
* @param nsq number of sub-quantizers (muliple of 2)
* @param codes packed codes array
* @param LUT packed look-up table
* @param accu array to store the results
*/
void accumulate_to_mem(
int nq,
size_t ntotal2,
int nsq,
const uint8_t* codes,
const uint8_t* LUT,
uint16_t* accu);

} // namespace faiss

0 comments on commit bcc9f32

Please sign in to comment.