Skip to content

Move driver and nvrtc cython and internal layers to new generator #1972

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

Open
mdboom wants to merge 18 commits into
base: main
Choose a base branch
from
Open

Move driver and nvrtc cython and internal layers to new generator #1972

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
f842321
Move driver and nvrtc cython and internal layers to new generator
mdboom Apr 24, 2026
992998a
Fix Cython interop tests
mdboom Apr 24, 2026
6b1d2d3
Handle headers differently
mdboom Apr 24, 2026
be5d972
Fix compilation
mdboom Apr 24, 2026
2d7c8dd
Make const match
mdboom Apr 24, 2026
f664c86
Fix compilation again
mdboom Apr 24, 2026
00ea580
Attempt to fix Windows build
mdboom Apr 25, 2026
9982e06
Fix Windows again
mdboom Apr 25, 2026
6bf0565
Fix not found error code
mdboom Apr 29, 2026
103c29f
Merge remote-tracking branch 'upstream/main' into driver-v2
mdboom Apr 29, 2026
591d101
Look up pointers in a different way
mdboom Apr 29, 2026
7119b96
Get first version right
mdboom Apr 29, 2026
0d7171c
Get first appearing versions correct
mdboom Apr 29, 2026
9bb9f5a
Fix weird special case
mdboom Apr 29, 2026
bcf5683
Merge remote-tracking branch 'upstream/main' into driver-v2
mdboom Jun 9, 2026
e2624e1
Merge branch 'main' into driver-v2
mdboom Jun 9, 2026
51bde1f
Fix Python 3.15 build
mdboom Jun 9, 2026
12cf111
Merge remote-tracking branch 'origin/driver-v2' into driver-v2
mdboom Jun 9, 2026
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
10 changes: 0 additions & 10 deletions .gitignore
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,31 +17,21 @@ __pycache__/
cache_driver
cache_runtime
cache_nvrtc
cuda_bindings/cuda/bindings/_lib/utils.pxi

# CUDA Python specific (auto-generated)
cuda_bindings/cuda/bindings/_bindings/cydriver.pxd
cuda_bindings/cuda/bindings/_bindings/cydriver.pyx
cuda_bindings/cuda/bindings/_bindings/cyruntime.pxd
cuda_bindings/cuda/bindings/_bindings/cyruntime.pyx
cuda_bindings/cuda/bindings/_bindings/cyruntime_ptds.pxd
cuda_bindings/cuda/bindings/_bindings/cyruntime_ptds.pyx
cuda_bindings/cuda/bindings/_internal/_nvml.pyx
cuda_bindings/cuda/bindings/_internal/cufile.pyx

@leofang leofang Jun 9, 2026

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

nit: we should add them here now that both are auto-gen'd?

Suggested change
cuda_bindings/cuda/bindings/_internal/cufile.pyx
cuda_bindings/cuda/bindings/_internal/driver.pyx
cuda_bindings/cuda/bindings/_internal/nvrtc.pyx
cuda_bindings/cuda/bindings/_internal/cufile.pyx

cuda_bindings/cuda/bindings/_internal/nvfatbin.pyx
cuda_bindings/cuda/bindings/_internal/nvjitlink.pyx
cuda_bindings/cuda/bindings/_internal/nvml.pyx
cuda_bindings/cuda/bindings/_internal/nvvm.pyx
cuda_bindings/cuda/bindings/_lib/utils.pxd
cuda_bindings/cuda/bindings/_lib/utils.pyx
cuda_bindings/cuda/bindings/cydriver.pxd
cuda_bindings/cuda/bindings/cydriver.pyx
cuda_bindings/cuda/bindings/cyruntime.pxd
cuda_bindings/cuda/bindings/cyruntime.pyx
cuda_bindings/cuda/bindings/cyruntime_functions.pxi
cuda_bindings/cuda/bindings/cyruntime_types.pxi
cuda_bindings/cuda/bindings/driver.pxd
cuda_bindings/cuda/bindings/driver.pyx
cuda_bindings/cuda/bindings/runtime.pxd
cuda_bindings/cuda/bindings/runtime.pyx
cuda_bindings/cuda/bindings/utils/_get_handle.pyx
Expand Down
7 changes: 0 additions & 7 deletions cuda_bindings/build_hooks.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -78,10 +78,6 @@ def _get_cuda_path() -> str:
# Header parsing helpers (called only from _build_cuda_bindings)

_REQUIRED_HEADERS = {
"driver": [
"cuda.h",
"cudaProfiler.h",
],
"runtime": [
"driver_types.h",
"vector_types.h",
Expand Down Expand Up @@ -449,9 +445,6 @@ def _cleanup_dst_files():
cuda_bindings_files = [f for f in cuda_bindings_files if "cufile" not in f]
sources_list = [
# private
(["cuda/bindings/_bindings/cydriver.pyx", "cuda/bindings/_bindings/loader.cpp"], None),
# cynvrtc.pyx is now in cuda/bindings/ (pre-generated by cybind),
# picked up by the cuda_bindings_files glob below.
(["cuda/bindings/_bindings/cyruntime.pyx"], static_runtime_libraries),
(["cuda/bindings/_bindings/cyruntime_ptds.pyx"], static_runtime_libraries),
# utils
Expand Down
Loading
Loading