Skip to content

perf: use pinned memory with preload_to_gpu#161

Merged
ilan-gold merged 15 commits intomainfrom
ig/pinned_memory
Mar 18, 2026
Merged

perf: use pinned memory with preload_to_gpu#161
ilan-gold merged 15 commits intomainfrom
ig/pinned_memory

Conversation

@ilan-gold
Copy link
Copy Markdown
Collaborator

Closes #83

@ilan-gold ilan-gold added the run-gpu-ci Signal that gpu ci should be run label Mar 16, 2026
@github-actions github-actions Bot removed the run-gpu-ci Signal that gpu ci should be run label Mar 16, 2026
@ilan-gold ilan-gold added the run-gpu-ci Signal that gpu ci should be run label Mar 16, 2026
@github-actions github-actions Bot removed the run-gpu-ci Signal that gpu ci should be run label Mar 16, 2026
@ilan-gold ilan-gold added the run-gpu-ci Signal that gpu ci should be run label Mar 16, 2026
@github-actions github-actions Bot removed the run-gpu-ci Signal that gpu ci should be run label Mar 16, 2026
@codecov
Copy link
Copy Markdown

codecov Bot commented Mar 16, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 92.79%. Comparing base (fac9795) to head (6011350).
⚠️ Report is 1 commits behind head on main.

Additional details and impacted files
@@            Coverage Diff             @@
##             main     #161      +/-   ##
==========================================
+ Coverage   92.10%   92.79%   +0.69%     
==========================================
  Files          11       11              
  Lines         937      944       +7     
==========================================
+ Hits          863      876      +13     
+ Misses         74       68       -6     
Files with missing lines Coverage Δ
src/annbatch/loader.py 92.81% <100.00%> (+0.16%) ⬆️

... and 1 file with indirect coverage changes

🚀 New features to boost your workflow:
  • ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

@ilan-gold ilan-gold requested a review from felix0097 March 16, 2026 16:03
Comment thread src/annbatch/loader.py Outdated
]
)

def get_kwargs(z: zarr.Array) -> dict:
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

why do you use a get_kwargs function here and not above?

Copy link
Copy Markdown
Collaborator Author

Choose a reason for hiding this comment

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

Because it's used in the list comprehension, but yea, good point for a refactor!

@ilan-gold ilan-gold added the run-gpu-ci Signal that gpu ci should be run label Mar 17, 2026
@github-actions github-actions Bot removed the run-gpu-ci Signal that gpu ci should be run label Mar 17, 2026
@ilan-gold ilan-gold requested a review from felix0097 March 17, 2026 13:46
Copy link
Copy Markdown
Collaborator

@felix0097 felix0097 left a comment

Choose a reason for hiding this comment

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

LGTM

@ilan-gold ilan-gold enabled auto-merge (squash) March 18, 2026 10:20
@ilan-gold ilan-gold added the run-gpu-ci Signal that gpu ci should be run label Mar 18, 2026
@github-actions github-actions Bot removed the run-gpu-ci Signal that gpu ci should be run label Mar 18, 2026
@ilan-gold ilan-gold merged commit b4bff4d into main Mar 18, 2026
16 of 17 checks passed
@ilan-gold ilan-gold deleted the ig/pinned_memory branch March 18, 2026 10:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Use pinned memory buffers from cupy if possible

2 participants