Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

perf: various fixes to improve shuffling performance at high scales #2710

Merged
merged 6 commits into from
Aug 19, 2024

Conversation

westonpace
Copy link
Contributor

@westonpace westonpace commented Aug 8, 2024

With this fix shuffling performance at 1M rows goes from ~2.8 seconds to ~0.8 seconds. At 1B rows we can shuffle in ~6 minutes.

@codecov-commenter
Copy link

codecov-commenter commented Aug 9, 2024

Codecov Report

Attention: Patch coverage is 91.47059% with 29 lines in your changes missing coverage. Please review.

Project coverage is 79.21%. Comparing base (7284521) to head (97c6628).
Report is 1 commits behind head on main.

Files Patch % Lines
rust/lance-index/src/vector/ivf/shuffler.rs 91.66% 8 Missing and 19 partials ⚠️
rust/lance-file/src/v2/writer.rs 87.50% 0 Missing and 2 partials ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main    #2710      +/-   ##
==========================================
+ Coverage   79.13%   79.21%   +0.07%     
==========================================
  Files         227      227              
  Lines       67398    67573     +175     
  Branches    67398    67573     +175     
==========================================
+ Hits        53338    53529     +191     
+ Misses      10956    10947       -9     
+ Partials     3104     3097       -7     
Flag Coverage Δ
unittests 79.21% <91.47%> (+0.07%) ⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@wjones127 wjones127 marked this pull request as draft August 13, 2024 18:17
@westonpace westonpace marked this pull request as ready for review August 19, 2024 12:45
Copy link
Contributor

@BubbleCal BubbleCal left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, just a NIT comment

rust/lance-index/src/vector/ivf/shuffler.rs Outdated Show resolved Hide resolved
westonpace and others added 2 commits August 19, 2024 07:47
Co-authored-by: BubbleCal <bubble-cal@outlook.com>
@westonpace westonpace merged commit 7a36b28 into lancedb:main Aug 19, 2024
21 of 22 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants