Merge branch 'main' into unsupported-nvml-hardware

Author: mdboom

ci/tools/install_gpu_driver.ps1

@@ -1,4 +1,4 @@
-# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # SPDX-License-Identifier: Apache-2.0
 
@@ -72,8 +72,13 @@ function Install-Driver {
         Write-Output "Unknown driver mode: $driver_mode"
         exit 1
     }
-    pnputil /disable-device /class Display
-    pnputil /enable-device /class Display
+    # Only restart NVIDIA display adapters, not other display devices (e.g. QEMU VGA)
+    $nvidia_devices = Get-PnpDevice -Class Display -FriendlyName "NVIDIA*"
+    foreach ($device in $nvidia_devices) {
+        Write-Output "Restarting device: $($device.FriendlyName) ($($device.InstanceId))"
+        pnputil /disable-device "$($device.InstanceId)"
+        pnputil /enable-device "$($device.InstanceId)"
+    }
     # Give it a minute to settle:
     Start-Sleep -Seconds 5
 }

cuda_bindings/pyproject.toml

@@ -56,6 +56,9 @@ Documentation = "https://nvidia.github.io/cuda-python/"
 [tool.setuptools.packages.find]
 include = ["cuda*"]
 
+[tool.setuptools.exclude-package-data]
+"*" = ["*.cpp"]
+
 [tool.setuptools.dynamic]
 readme = { file = ["DESCRIPTION.rst"], content-type = "text/x-rst" }
 

cuda_bindings/setup.py

@@ -356,6 +356,7 @@ def do_cythonize(extensions):
     return cythonize(
         extensions,
         nthreads=nthreads,
+        build_dir="build/cython",
         compiler_directives=compiler_directives,
         **extra_cythonize_kwargs,
     )

cuda_core/build_hooks.py

@@ -163,6 +163,7 @@ def get_sources(mod_name):
         ext_modules,
         verbose=True,
         language_level=3,
+        build_dir="build/cython",
         nthreads=nthreads,
         compiler_directives=compiler_directives,
         compile_time_env=compile_time_env,