-
Notifications
You must be signed in to change notification settings - Fork 231
Refactor device capability checks, fix prefetching bug on WSL #2014
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
Refactor device capability checks, fix prefetching bug on WSL #2014
Conversation
…ks to skip prefetching tests on WSL.
Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually. Contributors can view more details about this message here. |
/ok to test b6a1242 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Python changes look good.
/merge |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Approving build changes.
Description
Refactors device capability checks, add concurrent managed access checks to skip prefetching tests on WSL (not supported).
One motivation for this refactoring is that the previous names really didn't make much sense. We created the
runtime_async_alloc::is_supported()
to check for CUDA 11.2+ support for runtime async allocation. However, we tacked on other features likeis_export_handle_type_supported(cudaMemAllocationHandleType)
andis_hwdecompress_supported()
into the same struct. Those should have been their own features, not tied toruntime_async_alloc
. Most of the changes in this PR are structurally similar to that.Closes #2023.
Checklist