Skip to content

Commit

Permalink
Fix the code hang while caching nodes if search is run with a single …
Browse files Browse the repository at this point in the history
…thread (#590)

* Fixing other places with the single search thread hang issue

* Fixing other places with the single search thread hang issue

* Adding overwrite option to delete old versions

* Add a run id to files to avoid naming conflicts

---------

Co-authored-by: Gopal Srinivasa <[email protected]>
  • Loading branch information
gopalrs and gopal-msr authored Oct 15, 2024
1 parent bae427e commit 6f2691c
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 6 deletions.
3 changes: 2 additions & 1 deletion .github/workflows/build-python-pdoc.yml
Original file line number Diff line number Diff line change
Expand Up @@ -47,7 +47,8 @@ jobs:
with:
name: dependencies
path: |
dependencies_documentation.txt
${{ github.run_id }}-dependencies_documentation.txt
overwrite: true
- name: Archive documentation artifacts
uses: actions/upload-artifact@v4
with:
Expand Down
5 changes: 0 additions & 5 deletions src/pq_flash_index.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -365,11 +365,6 @@ void PQFlashIndex<T, LabelT>::cache_bfs_levels(uint64_t num_nodes_to_cache, std:
}
diskann::cout << "Caching " << num_nodes_to_cache << "..." << std::endl;

// borrow thread data
ScratchStoreManager<SSDThreadData<T>> manager(this->_thread_data);
auto this_thread_data = manager.scratch_space();
IOContext &ctx = this_thread_data->ctx;

std::unique_ptr<tsl::robin_set<uint32_t>> cur_level, prev_level;
cur_level = std::make_unique<tsl::robin_set<uint32_t>>();
prev_level = std::make_unique<tsl::robin_set<uint32_t>>();
Expand Down

0 comments on commit 6f2691c

Please sign in to comment.