diff --git a/pkgs/development/python-modules/cuda-pathfinder/default.nix b/pkgs/development/python-modules/cuda-pathfinder/default.nix new file mode 100644 index 000000000000..1ca842a476d4 --- /dev/null +++ b/pkgs/development/python-modules/cuda-pathfinder/default.nix @@ -0,0 +1,52 @@ +{ + lib, + buildPythonPackage, + fetchFromGitHub, + + # build-system + setuptools, + setuptools-scm, + + # tests + pytest-mock, + pytestCheckHook, +}: + +buildPythonPackage (finalAttrs: { + pname = "cuda-pathfinder"; + version = "1.4.0"; + pyproject = true; + + src = fetchFromGitHub { + owner = "NVIDIA"; + repo = "cuda-python"; + tag = "cuda-pathfinder-v${finalAttrs.version}"; + hash = "sha256-Bsou6vLyMBNbVMPT4vtnWpoi05lXG6pjhuee6Hg/Mm8="; + }; + + sourceRoot = "${finalAttrs.src.name}/cuda_pathfinder"; + + build-system = [ + setuptools + setuptools-scm + ]; + + pythonImportsCheck = [ + "cuda" + "cuda.pathfinder" + ]; + + nativeCheckInputs = [ + pytest-mock + pytestCheckHook + ]; + + meta = { + description = "one-stop solution for locating CUDA components"; + homepage = "https://github.com/NVIDIA/cuda-python/tree/main/cuda_pathfinder"; + changelog = "https://nvidia.github.io/cuda-python/cuda-pathfinder/${finalAttrs.version}/release/${finalAttrs.version}-notes.html"; + license = lib.licenses.asl20; + maintainers = with lib.maintainers; [ GaetanLepage ]; + platforms = lib.platforms.linux; + }; +}) diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 1d89c5e30dd5..5749d69bc2fd 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -3386,6 +3386,8 @@ self: super: with self; { cuda-bindings = callPackage ../development/python-modules/cuda-bindings { }; + cuda-pathfinder = callPackage ../development/python-modules/cuda-pathfinder { }; + cupy = callPackage ../development/python-modules/cupy { cudaPackages = # CuDNN 9 is not supported: