Unverified Commit fee1f07b authored by Nick Cao's avatar Nick Cao Committed by GitHub
Browse files

python3Packages.cuda-pathfinder: 1.4.0 -> 1.5.3;... 

python3Packages.cuda-pathfinder: 1.4.0 -> 1.5.3; python3Packages.cuda-bindings: 12.8.0 -> 12.9.6 (#511746)
parents 560d577d 26c26837

pkgs/development/python-modules/cuda-bindings/default.nix

+15 −17
Original line number Diff line number Diff line
@@ -7,12 +7,16 @@ 


  # build-system

  cython,

  setuptools,

  pyclibrary,

  setuptools,

  setuptools-scm,



  # env

  symlinkJoin,



  # dependencies

  cuda-pathfinder,



  # tests

  numpy,

  pytestCheckHook,
@@ -23,14 +27,15 @@ 


buildPythonPackage (finalAttrs: {

  pname = "cuda-bindings";

  version = "12.8.0";

  version = "12.9.6";

  pyproject = true;

  __structuredAttrs = true;



  src = fetchFromGitHub {

    owner = "NVIDIA";

    repo = "cuda-python";

    tag = "v${finalAttrs.version}";

    hash = "sha256-7e9w70KkC6Pcvyu6Cwt5Asrc3W9TgsjiGvArRTer6Oc=";

    hash = "sha256-uRv27h2b6wXC8oOf5k2KxZ0bUFNvNu6XO05FBbJcU1k=";

  };



  sourceRoot = "${finalAttrs.src.name}/cuda_bindings";
@@ -48,9 +53,6 @@ buildPythonPackage (finalAttrs: { 
    in

    ''

      substituteInPlace cuda/bindings/_internal/nvjitlink_linux.pyx \

        --replace-fail \

          'so_name = "libnvJitLink.so"' \

          'so_name = "${lib.getLib cudaPackages.libnvjitlink}/lib/libnvJitLink.so"' \

        --replace-fail \

          "handle = dlopen('libcuda.so.1'" \

          "handle = dlopen('${libCudaPath}/lib/libcuda.so.1'"
@@ -59,16 +61,6 @@ buildPythonPackage (finalAttrs: { 
        --replace-fail \

          "path = 'libcuda.so.1'" \

          "path = '${libCudaPath}/lib/libcuda.so.1'"



      substituteInPlace cuda/bindings/_bindings/cynvrtc.pyx.in \

        --replace-fail \

          "dlfcn.dlopen('libnvrtc.so.12'" \

          "dlfcn.dlopen('${lib.getLib cudaPackages.cuda_nvrtc}/lib/libnvrtc.so.12'"



      substituteInPlace cuda/bindings/_lib/cyruntime/cyruntime.pyx.in \

        --replace-fail \

          "dlfcn.dlopen('libcudart.so.12'" \

          "dlfcn.dlopen('${lib.getLib cudaPackages.cuda_cudart}/lib/libcudart.so.12'"

    '';



  preBuild = ''
@@ -79,6 +71,7 @@ buildPythonPackage (finalAttrs: { 
    cython

    pyclibrary

    setuptools

    setuptools-scm

  ];



  env = {
@@ -94,6 +87,11 @@ buildPythonPackage (finalAttrs: { 


  buildInputs = [

    cudaPackages.cuda_nvcc # crt/host_defines.h

    cudaPackages.libcufile # cufile.h

  ];



  dependencies = [

    cuda-pathfinder

  ];



  pythonImportsCheck = [
@@ -133,7 +131,7 @@ buildPythonPackage (finalAttrs: { 
  };



  meta = {

    description = "CUDA Python: Performance meets Productivity";

    description = "Standard set of low-level interfaces, providing access to the CUDA host APIs from Python";

    homepage = "https://github.com/NVIDIA/cuda-python/tree/main/cuda_bindings";

    changelog = "https://nvidia.github.io/cuda-python/${finalAttrs.version}/release/${finalAttrs.version}-notes.html";

    license = lib.licenses.unfreeRedistributable; # NVIDIA Proprietary Software

pkgs/development/python-modules/cuda-pathfinder/default.nix

+3 −2
Original line number Diff line number Diff line
@@ -14,14 +14,15 @@ 


buildPythonPackage (finalAttrs: {

  pname = "cuda-pathfinder";

  version = "1.4.0";

  version = "1.5.3";

  pyproject = true;

  __structuredAttrs = true;



  src = fetchFromGitHub {

    owner = "NVIDIA";

    repo = "cuda-python";

    tag = "cuda-pathfinder-v${finalAttrs.version}";

    hash = "sha256-Bsou6vLyMBNbVMPT4vtnWpoi05lXG6pjhuee6Hg/Mm8=";

    hash = "sha256-Tj+0p+nIsOl2pMpKAUpdZ3nIcQ0kHWrPi6Qeu14oMRQ=";

  };



  sourceRoot = "${finalAttrs.src.name}/cuda_pathfinder";