diff --git a/.gitignore b/.gitignore index f80e5c682..faf51669b 100644 --- a/.gitignore +++ b/.gitignore @@ -378,3 +378,5 @@ gperftools # Rust rust/target + +python/src/*.so \ No newline at end of file diff --git a/include/ann_exception.h b/include/ann_exception.h index 6b81373c1..55f069b0e 100644 --- a/include/ann_exception.h +++ b/include/ann_exception.h @@ -6,6 +6,7 @@ #include #include #include "windows_customizations.h" +#include #ifndef _WINDOWS #define __FUNCSIG__ __PRETTY_FUNCTION__ diff --git a/src/pq_flash_index.cpp b/src/pq_flash_index.cpp index d9ad50617..6c004023d 100644 --- a/src/pq_flash_index.cpp +++ b/src/pq_flash_index.cpp @@ -1271,7 +1271,10 @@ void PQFlashIndex::cached_beam_search(const T *query1, const uint64_t const uint32_t io_limit, const bool use_reorder_data, QueryStats *stats) { - + // printf("cached_beam_search\n"); + // diskann::cout << "cached_beam_search" << std::endl; + // printf("cached_beam_search\n"); + // diskann::cout << "cached_beam_search" << std::endl; uint64_t num_sector_per_nodes = DIV_ROUND_UP(_max_node_len, defaults::SECTOR_LEN); if (beam_width > num_sector_per_nodes * defaults::MAX_N_SECTOR_READS) throw ANNException("Beamwidth can not be higher than defaults::MAX_N_SECTOR_READS", -1, __FUNCSIG__, __FILE__, @@ -1529,6 +1532,7 @@ void PQFlashIndex::cached_beam_search(const T *query1, const uint64_t cmps++; float dist = dist_scratch[m]; Neighbor nn(id, dist); + // expand the neighbor here retset.insert(nn); } }