Skip to content

Prerelease NVML bindings #1284

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.

Sign up for GitHub

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

Draft
wants to merge 9 commits into
base: main
Choose a base branch
from
+45,596 −792
Draft

Prerelease NVML bindings #1284

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
a88c6be
First full pass at NVML bindings
mdboom Nov 24, 2025
55e0fc9
More tests
mdboom Nov 24, 2025
ac41b42
More tests
mdboom Nov 25, 2025
2754081
Whitespace fixes
mdboom Dec 1, 2025
586a452
Make NVML work on Windows
mdboom Dec 1, 2025
315e12d
Skip another test on Windows
mdboom Dec 1, 2025
a176ebd
Free-threading fixes
mdboom Dec 1, 2025
ad4ffc2
Fix compilation
mdboom Dec 1, 2025
f932f0d
Fix free-threading
mdboom Dec 1, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .pre-commit-config.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,7 @@ repos:
rev: "3e8a8703264a2f4a69428a0aa4dcb512790b2c8c" # frozen: v6.0.0
hooks:
- id: check-added-large-files
exclude: cuda_bindings/cuda/bindings/nvml.pyx
- id: check-case-conflict
- id: check-docstring-first
- id: check-merge-conflict
Expand Down
361 changes: 361 additions & 0 deletions cuda_bindings/cuda/bindings/_internal/nvml.pxd
View file
Open in desktop

Large diffs are not rendered by default.

7,421 changes: 7,421 additions & 0 deletions cuda_bindings/cuda/bindings/_internal/nvml_linux.pyx
View file
Open in desktop

Large diffs are not rendered by default.

6,071 changes: 6,071 additions & 0 deletions cuda_bindings/cuda/bindings/_internal/nvml_windows.pyx
View file
Open in desktop

Large diffs are not rendered by default.

4 changes: 2 additions & 2 deletions cuda_bindings/cuda/bindings/cufile.pxd
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ ctypedef CUFileArrayConfigParameter_t _ArrayConfigParameter
###############################################################################

cpdef intptr_t handle_register(intptr_t descr) except? 0
cpdef void handle_deregister(intptr_t fh) except*
cpdef void handle_deregister(intptr_t fh)except*
cpdef buf_register(intptr_t buf_ptr_base, size_t length, int flags)
cpdef buf_deregister(intptr_t buf_ptr_base)
cpdef driver_open()
Expand All @@ -58,7 +58,7 @@ cpdef intptr_t batch_io_set_up(unsigned nr) except? 0
cpdef batch_io_submit(intptr_t batch_idp, unsigned nr, intptr_t iocbp, unsigned int flags)
cpdef batch_io_get_status(intptr_t batch_idp, unsigned min_nr, intptr_t nr, intptr_t iocbp, intptr_t timeout)
cpdef batch_io_cancel(intptr_t batch_idp)
cpdef void batch_io_destroy(intptr_t batch_idp) except*
cpdef void batch_io_destroy(intptr_t batch_idp)except*
cpdef read_async(intptr_t fh, intptr_t buf_ptr_base, intptr_t size_p, intptr_t file_offset_p, intptr_t buf_ptr_offset_p, intptr_t bytes_read_p, intptr_t stream)
cpdef write_async(intptr_t fh, intptr_t buf_ptr_base, intptr_t size_p, intptr_t file_offset_p, intptr_t buf_ptr_offset_p, intptr_t bytes_written_p, intptr_t stream)
cpdef stream_register(intptr_t stream, unsigned flags)
Expand Down
Loading
Loading