[FEA]: Add CUDA driver version query to cuda.pathfinder (private API)#1953
Merged
rwgk merged 16 commits intoNVIDIA:mainfrom Apr 24, 2026
Merged
[FEA]: Add CUDA driver version query to cuda.pathfinder (private API)#1953rwgk merged 16 commits intoNVIDIA:mainfrom
cuda.pathfinder (private API)#1953rwgk merged 16 commits intoNVIDIA:mainfrom
Conversation
Break out CUDA driver version querying into a standalone internal utility so it can be reused independently from compatibility checks, and cover the ctypes loader paths with focused tests. Made-with: Cursor
Treat the Windows WinDLL path as the normal runtime case and keep the focused tests aligned with the stricter driver-loader invariants. Made-with: Cursor
Wrap the encoded cuDriverGetVersion() result in a DriverVersion dataclass so callers can use major and minor fields directly while retaining a low-level integer helper for loader-focused tests. Made-with: Cursor
Cover query_driver_version() alongside the driver library loading tests and reuse the existing strictness mode so host-specific failures still surface cleanly in all_must_work mode. Made-with: Cursor
Drop the non-Windows loader mock now that a real driver-version test covers the Linux success path, while keeping the Windows branch and failure-path unit coverage. Made-with: Cursor
Use DriverCudaVersion for clearer pairing with the planned release-version type while keeping the existing driver info API behavior unchanged. Made-with: Cursor
Query nvmlSystemGetDriverVersion() through pathfinder's driver library loading path and add a minimal real test so the implementation is preserved as a future reference. Made-with: Cursor
Step back from the exploratory NVML-based release-version query for now because it adds non-trivial complexity and a new dependency surface without a current pathfinder need, while keeping the reference implementation in history if we need it later. Made-with: Cursor
Document that DriverCudaVersion matches the CUDA Version shown by nvidia-smi rather than the graphics driver release, so the dataclass name reads clearly in context. Made-with: Cursor
Expose DriverCudaVersion, QueryDriverCudaVersionError, and query_driver_cuda_version publicly, and align the internal naming, caching, docs, and test coverage around the CUDA-specific driver version query. Made-with: Cursor
Protect the new top-level driver-info re-exports so internal-only test coverage does not miss a broken `cuda.pathfinder` plumbing layer. Made-with: Cursor
rwgk
added
P0
Contributor
|
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. |
Contributor
Author
|
/ok to test |
This comment has been minimized.
This comment has been minimized.
leofang
requested changes
Apr 21, 2026
Stop exposing the new driver info helper through cuda.pathfinder while keeping the internal implementation and internal test coverage in place. Made-with: Cursor
Contributor
Author
|
/ok to test |
Contributor
Author
|
|
rwgk
changed the title
cuda.pathfindercuda.pathfinder (private API)
leofang
approved these changes
Apr 24, 2026
Comment on lines
+26
to
+28
| The name ``DriverCudaVersion`` is intentionally specific: this dataclass | ||
| models the version shown as ``CUDA Version`` in ``nvidia-smi``, not the | ||
| graphics driver release shown as ``Driver Version``. |
Member
There was a problem hiding this comment.
In other words, this class records the UMD version, not the KMD version
Contributor
Author
There was a problem hiding this comment.
Thanks! I added commit 0d7c8f8 to add that information on.
Spell out that `cuDriverGetVersion()` reports the CUDA-facing user-mode driver (UMD) version rather than the kernel-mode driver (KMD) package version so the `nvidia-smi` comparison is less ambiguous. Made-with: Cursor
rwgk
added a commit
to rwgk/cuda-python
that referenced
this pull request
Apr 24, 2026
|
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
2 participants
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
xref: #1038, #1936
cuda.pathfinder._utils.driver_infohelper backed bycuDriverGetVersion()socuda.pathfindercan query the CUDA-facing driver versionDriverCudaVersion,QueryDriverCudaVersionError, and cachedquery_driver_cuda_version(), with naming and docstrings that make it explicit this matchesnvidia-smi'sCUDA Versionre-export the new API fromcuda.pathfinder, document it incuda_pathfinder/docs/source/api.rst, and add focused unit, real-driver, and public-import regression testsAll new
INFOlines are identical across all CI test logs:https://github.com/NVIDIA/cuda-python/actions/runs/24639785901/attempts/1?pr=1953