python3Packages.cuda-core: init at 1.0.1 (#538483)

2026-07-05 20:13:45 +00:00
parent fe0a117b3b 9a58143dca
commit d5e64fa9b7
2 changed files with 138 additions and 0 deletions
@@ -0,0 +1,136 @@
+{
+  lib,
+  config,
+  buildPythonPackage,
+  cudaPackages,
+  fetchFromGitHub,
+  symlinkJoin,
+
+  # build-system
+  cuda-bindings,
+  cuda-pathfinder,
+  cython,
+  setuptools,
+  setuptools-scm,
+
+  # dependencies
+  numpy,
+
+  # tests
+  cffi,
+  cloudpickle,
+  psutil,
+  pytestCheckHook,
+
+  # passthru
+  cuda-core,
+
+  cudaSupport ? config.cudaSupport,
+}:
+buildPythonPackage.override { stdenv = cudaPackages.backendStdenv; } (finalAttrs: {
+  pname = "cuda-core";
+  version = "1.0.1";
+  pyproject = true;
+  __structuredAttrs = true;
+
+  src = fetchFromGitHub {
+    owner = "NVIDIA";
+    repo = "cuda-python";
+    tag = "cuda-core-v${finalAttrs.version}";
+    hash = "sha256-SRy/hnOzzb4wUCLOre4k326RNhYI0650XzC8Dc9kf/M=";
+  };
+
+  sourceRoot = "${finalAttrs.src.name}/cuda_core";
+
+  env = {
+    CUDA_HOME = symlinkJoin {
+      name = "cuda-redist";
+      paths =
+        with cudaPackages;
+        [
+          # Used to detect CUDA MAJOR VERSION
+          (lib.getInclude cuda_cudart)
+        ]
+        ++ lib.optionals finalAttrs.doInstallCheck [
+          # Compilation tests include <cuda/atomic> etc. (found via $CUDA_HOME/include)
+          (lib.getInclude cccl)
+        ];
+    };
+  };
+
+  build-system = [
+    cuda-bindings
+    cuda-pathfinder
+    cython
+    setuptools
+    setuptools-scm
+  ];
+
+  nativeBuildInputs = [
+    cudaPackages.cuda_nvcc
+  ];
+
+  buildInputs = [
+    cudaPackages.cuda_nvrtc # nvrtc.h
+    cudaPackages.cuda_profiler_api # cudaProfiler.h
+  ];
+
+  preBuild = ''
+    export CUDA_PYTHON_PARALLEL_LEVEL=$NIX_BUILD_CORES
+  '';
+
+  dependencies = [
+    cuda-pathfinder
+    numpy
+  ];
+
+  pythonImportsCheck = [ "cuda.core" ];
+
+  preCheck = ''
+    rm -rf cuda/core
+  '';
+
+  nativeCheckInputs = [
+    cffi
+    cloudpickle
+    psutil
+    pytestCheckHook
+  ];
+
+  disabledTests = [
+    # Fail inside the sandbox:
+    #   cuda.bindings.nvml.UnknownError: Unknown Error
+    #   hwloc/linux: failed to find sysfs cpu topology directory, aborting linux discovery.
+    "test_affinity"
+    "test_device_cpu_affinity"
+    "test_device_pci_bus_id"
+    "test_device_pci_info"
+    "test_to_cuda_device"
+    "test_to_system_device"
+  ];
+
+  disabledTestPaths = [
+    # AssertionError (tries to run an external python process that imports `cuda`)
+    # ModuleNotFoundError: No module named 'cuda'
+    "tests/test_rlcompleter_patch.py"
+  ];
+
+  # Tests require a GPU
+  doCheck = false;
+
+  passthru.gpuCheck = cuda-core.overridePythonAttrs {
+    requiredSystemFeatures = [ "cuda" ];
+    doCheck = true;
+  };
+
+  meta = {
+    description = "Pythonic interface to the CUDA runtime";
+    homepage = "https://nvidia.github.io/cuda-python/cuda-core/latest";
+    downloadPage = "https://github.com/NVIDIA/cuda-python/tree/main/cuda_core";
+    license = lib.licenses.asl20;
+    maintainers = with lib.maintainers; [
+      GaetanLepage
+    ];
+    broken = !cudaSupport;
+  };
+})
@@ -3775,6 +3775,8 @@ self: super: with self; {

  cuda-bindings = callPackage ../development/python-modules/cuda-bindings { };

+  cuda-core = callPackage ../development/python-modules/cuda-core { };
+
  cuda-pathfinder = callPackage ../development/python-modules/cuda-pathfinder { };

  cuda-tile = callPackage ../development/python-modules/cuda-tile { };