Skip to content

Commit 532fdc9

Browse files
mdboomleofang
andcommitted
Move driver and nvrtc cython and internal layers to new generator (#1972)
* Move driver and nvrtc cython and internal layers to new generator * Fix Cython interop tests * Handle headers differently * Fix compilation * Make const match * Fix compilation again * Attempt to fix Windows build * Fix Windows again * Fix not found error code * Look up pointers in a different way * Get first version right * Get first appearing versions correct * Fix weird special case * Fix exception signature to match what was there before * Also fix exception declaration in nvrtc * Fix PTDS and add PTDS testing * Update .gitignore Co-authored-by: Leo Fang <leof@nvidia.com> --------- Co-authored-by: Leo Fang <leof@nvidia.com>
1 parent c5c1db1 commit 532fdc9

34 files changed

Lines changed: 33527 additions & 40473 deletions

.gitignore

Lines changed: 4 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -16,27 +16,22 @@ __pycache__/
1616
cache_driver
1717
cache_runtime
1818
cache_nvrtc
19-
cuda_bindings/cuda/bindings/_lib/utils.pxi
2019

2120
# CUDA Python specific (auto-generated)
22-
cuda_bindings/cuda/bindings/_bindings/cydriver.pxd
23-
cuda_bindings/cuda/bindings/_bindings/cydriver.pyx
2421
cuda_bindings/cuda/bindings/_bindings/cyruntime.pxd
2522
cuda_bindings/cuda/bindings/_bindings/cyruntime.pyx
2623
cuda_bindings/cuda/bindings/_bindings/cyruntime_ptds.pxd
2724
cuda_bindings/cuda/bindings/_bindings/cyruntime_ptds.pyx
25+
cuda_bindings/cuda/bindings/_internal/driver.pyx
26+
cuda_bindings/cuda/bindings/_internal/nvrtc.pyx
27+
cuda_bindings/cuda/bindings/_internal/cufile.pyx
28+
cuda_bindings/cuda/bindings/_internal/nvfatbin.pyx
2829
cuda_bindings/cuda/bindings/_internal/nvjitlink.pyx
2930
cuda_bindings/cuda/bindings/_internal/nvvm.pyx
30-
cuda_bindings/cuda/bindings/_lib/utils.pxd
31-
cuda_bindings/cuda/bindings/_lib/utils.pyx
32-
cuda_bindings/cuda/bindings/cydriver.pxd
33-
cuda_bindings/cuda/bindings/cydriver.pyx
3431
cuda_bindings/cuda/bindings/cyruntime.pxd
3532
cuda_bindings/cuda/bindings/cyruntime.pyx
3633
cuda_bindings/cuda/bindings/cyruntime_functions.pxi
3734
cuda_bindings/cuda/bindings/cyruntime_types.pxi
38-
cuda_bindings/cuda/bindings/driver.pxd
39-
cuda_bindings/cuda/bindings/driver.pyx
4035
cuda_bindings/cuda/bindings/runtime.pxd
4136
cuda_bindings/cuda/bindings/runtime.pyx
4237
cuda_bindings/cuda/bindings/utils/_get_handle.pyx

cuda_bindings/build_hooks.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -47,10 +47,6 @@ def _get_cuda_paths() -> list[str]:
4747
# Header parsing helpers (called only from _build_cuda_bindings)
4848

4949
_REQUIRED_HEADERS = {
50-
"driver": [
51-
"cuda.h",
52-
"cudaProfiler.h",
53-
],
5450
"runtime": [
5551
"driver_types.h",
5652
"vector_types.h",
@@ -418,9 +414,6 @@ def _cleanup_dst_files():
418414
cuda_bindings_files = [f for f in cuda_bindings_files if "cufile" not in f]
419415
sources_list = [
420416
# private
421-
(["cuda/bindings/_bindings/cydriver.pyx", "cuda/bindings/_bindings/loader.cpp"], None),
422-
# cynvrtc.pyx is now in cuda/bindings/ (pre-generated by cybind),
423-
# picked up by the cuda_bindings_files glob below.
424417
(["cuda/bindings/_bindings/cyruntime.pyx"], static_runtime_libraries),
425418
(["cuda/bindings/_bindings/cyruntime_ptds.pyx"], static_runtime_libraries),
426419
# utils

cuda_bindings/cuda/bindings/_bindings/cydriver.pyx.in

Lines changed: 0 additions & 16278 deletions
This file was deleted.

0 commit comments

Comments
 (0)