Share managed-memory test helper

Move the managed-memory skip logic into cuda_python_test_helpers and point
bindings/core tests at the shared module, with path bootstrapping to prefer
in-repo helpers. This avoids relying on bindings test helpers that are absent
in 12.9.x wheels.

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: rwgk

cuda_bindings/tests/conftest.py

@@ -1,6 +1,14 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
+import pathlib
+import sys
+
+helpers_root = pathlib.Path(__file__).resolve().parents[2] / "cuda_python_test_helpers"
+if helpers_root.is_dir() and str(helpers_root) not in sys.path:
+    # Prefer the in-repo helpers over any installed copy.
+    sys.path.insert(0, str(helpers_root))
+
 import cuda.bindings.driver as cuda
 import pytest
 

cuda_bindings/tests/test_cuda.py

@@ -10,7 +10,7 @@
 import numpy as np
 import pytest
 from cuda.bindings import driver
-from cuda.bindings._test_helpers.managed_memory import managed_memory_skip_reason
+from cuda_python_test_helpers.managed_memory import managed_memory_skip_reason
 
 
 def driverVersionLessThan(target):

cuda_core/tests/conftest.py

@@ -6,7 +6,7 @@
 
 import helpers
 import pytest
-from cuda.bindings._test_helpers.managed_memory import managed_memory_skip_reason
+from cuda_python_test_helpers.managed_memory import managed_memory_skip_reason
 
 try:
     from cuda.bindings import driver

cuda_core/tests/helpers/__init__.py

@@ -22,12 +22,12 @@
             CCCL_INCLUDE_PATHS = (path,) + CCCL_INCLUDE_PATHS
 
 
-try:
-    from cuda_python_test_helpers import *  # noqa: F403
-except ImportError:
-    # Import shared platform helpers for tests across repos
-    sys.path.insert(0, str(pathlib.Path(__file__).resolve().parents[3] / "cuda_python_test_helpers"))
-    from cuda_python_test_helpers import *  # noqa: F403
+helpers_root = pathlib.Path(__file__).resolve().parents[3] / "cuda_python_test_helpers"
+if helpers_root.is_dir() and str(helpers_root) not in sys.path:
+    # Prefer the in-repo helpers over any installed copy.
+    sys.path.insert(0, str(helpers_root))
+
+from cuda_python_test_helpers import *  # noqa: E402, F403
 
 
 @functools.cache

…bindings/_test_helpers/managed_memory.py …da_python_test_helpers/managed_memory.pycuda_bindings/cuda/bindings/_test_helpers/managed_memory.py renamed to cuda_python_test_helpers/cuda_python_test_helpers/managed_memory.py cuda_bindings/cuda/bindings/_test_helpers/managed_memory.py renamed to cuda_python_test_helpers/cuda_python_test_helpers/managed_memory.py

@@ -1,5 +1,5 @@
 # SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
-# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
+# SPDX-License-Identifier: Apache-2.0
 
 from __future__ import annotations
 
@@ -8,9 +8,9 @@
 import pytest
 
 try:
-    from cuda.bindings import driver
+    from cuda.bindings import driver  # type: ignore
 except Exception:
-    from cuda import cuda as driver
+    from cuda import cuda as driver  # type: ignore
 
 
 def _resolve_device_id(device) -> int: