diff --git a/faiss/IndexPQ.cpp b/faiss/IndexPQ.cpp index 7037c24322..8193e78b17 100644 --- a/faiss/IndexPQ.cpp +++ b/faiss/IndexPQ.cpp @@ -159,16 +159,16 @@ void IndexPQ::search( FAISS_THROW_IF_NOT(is_trained); const SearchParametersPQ* params = nullptr; - Search_type_t search_type_2 = this->search_type; + Search_type_t param_search_type = this->search_type; if (iparams) { params = dynamic_cast(iparams); FAISS_THROW_IF_NOT_MSG(params, "invalid search params"); FAISS_THROW_IF_NOT_MSG(!params->sel, "selector not supported"); - search_type_2 = params->search_type; + param_search_type = params->search_type; } - if (search_type_2 == ST_PQ) { // Simple PQ search + if (param_search_type == ST_PQ) { // Simple PQ search if (metric_type == METRIC_L2) { float_maxheap_array_t res = { @@ -183,10 +183,10 @@ void IndexPQ::search( indexPQ_stats.ncode += n * ntotal; } else if ( - search_type_2 == ST_polysemous || - search_type_2 == ST_polysemous_generalize) { + param_search_type == ST_polysemous || + param_search_type == ST_polysemous_generalize) { FAISS_THROW_IF_NOT(metric_type == METRIC_L2); - int polysemous_ht_2 = + int param_polysemous_ht = params ? params->polysemous_ht : this->polysemous_ht; search_core_polysemous( n, @@ -194,8 +194,8 @@ void IndexPQ::search( k, distances, labels, - polysemous_ht_2, - search_type_2 == ST_polysemous_generalize); + param_polysemous_ht, + param_search_type == ST_polysemous_generalize); } else { // code-to-code distances @@ -215,7 +215,7 @@ void IndexPQ::search( } } - if (search_type_2 == ST_SDC) { + if (param_search_type == ST_SDC) { float_maxheap_array_t res = { size_t(n), size_t(k), labels, distances}; @@ -227,7 +227,7 @@ void IndexPQ::search( int_maxheap_array_t res = { size_t(n), size_t(k), labels, idistances.get()}; - if (search_type_2 == ST_HE) { + if (param_search_type == ST_HE) { hammings_knn_hc( &res, q_codes.get(), @@ -236,7 +236,7 @@ void IndexPQ::search( pq.code_size, true); - } else if (search_type_2 == ST_generalized_HE) { + } else if (param_search_type == ST_generalized_HE) { generalized_hammings_knn_hc( &res, q_codes.get(), @@ -322,13 +322,13 @@ void IndexPQ::search_core_polysemous( idx_t k, float* distances, idx_t* labels, - int polysemous_ht_2, + int param_polysemous_ht, bool generalized_hamming) const { FAISS_THROW_IF_NOT(k > 0); FAISS_THROW_IF_NOT(pq.nbits == 8); - if (polysemous_ht_2 == 0) { - polysemous_ht_2 = pq.nbits * pq.M + 1; + if (param_polysemous_ht == 0) { + param_polysemous_ht = pq.nbits * pq.M + 1; } // PQ distance tables @@ -374,7 +374,7 @@ void IndexPQ::search_core_polysemous( k, heap_dis, heap_ids, - polysemous_ht_2); + param_polysemous_ht); } else { // generalized hamming switch (pq.code_size) { @@ -387,7 +387,7 @@ void IndexPQ::search_core_polysemous( k, \ heap_dis, \ heap_ids, \ - polysemous_ht_2); \ + param_polysemous_ht); \ break; DISPATCH(8) DISPATCH(16) @@ -401,7 +401,7 @@ void IndexPQ::search_core_polysemous( k, heap_dis, heap_ids, - polysemous_ht_2); + param_polysemous_ht); } else { bad_code_size++; }