From f3ceb798536023105280e51bf710c4666798bc0d Mon Sep 17 00:00:00 2001 From: Connor Baker Date: Tue, 23 Sep 2025 17:12:24 +0000 Subject: [PATCH] cudaPackages.nvpl: init at 25.5 Signed-off-by: Connor Baker --- .../_cuda/manifests/nvpl/README.md | 5 + .../_cuda/manifests/nvpl/redistrib_25.5.json | 101 ++++++++++++++++++ .../cuda-modules/packages/nvpl_blas.nix | 18 ++++ .../cuda-modules/packages/nvpl_common.nix | 16 +++ .../cuda-modules/packages/nvpl_fft.nix | 18 ++++ .../cuda-modules/packages/nvpl_lapack.nix | 20 ++++ .../cuda-modules/packages/nvpl_rand.nix | 18 ++++ .../cuda-modules/packages/nvpl_scalapack.nix | 18 ++++ .../cuda-modules/packages/nvpl_sparse.nix | 19 ++++ .../cuda-modules/packages/nvpl_tensor.nix | 20 ++++ pkgs/top-level/cuda-packages.nix | 4 + 11 files changed, 257 insertions(+) create mode 100644 pkgs/development/cuda-modules/_cuda/manifests/nvpl/README.md create mode 100644 pkgs/development/cuda-modules/_cuda/manifests/nvpl/redistrib_25.5.json create mode 100644 pkgs/development/cuda-modules/packages/nvpl_blas.nix create mode 100644 pkgs/development/cuda-modules/packages/nvpl_common.nix create mode 100644 pkgs/development/cuda-modules/packages/nvpl_fft.nix create mode 100644 pkgs/development/cuda-modules/packages/nvpl_lapack.nix create mode 100644 pkgs/development/cuda-modules/packages/nvpl_rand.nix create mode 100644 pkgs/development/cuda-modules/packages/nvpl_scalapack.nix create mode 100644 pkgs/development/cuda-modules/packages/nvpl_sparse.nix create mode 100644 pkgs/development/cuda-modules/packages/nvpl_tensor.nix diff --git a/pkgs/development/cuda-modules/_cuda/manifests/nvpl/README.md b/pkgs/development/cuda-modules/_cuda/manifests/nvpl/README.md new file mode 100644 index 000000000000..1a338d1c2461 --- /dev/null +++ b/pkgs/development/cuda-modules/_cuda/manifests/nvpl/README.md @@ -0,0 +1,5 @@ +# nvpl + +Link: + +Requirements: diff --git a/pkgs/development/cuda-modules/_cuda/manifests/nvpl/redistrib_25.5.json b/pkgs/development/cuda-modules/_cuda/manifests/nvpl/redistrib_25.5.json new file mode 100644 index 000000000000..d7739f946a1c --- /dev/null +++ b/pkgs/development/cuda-modules/_cuda/manifests/nvpl/redistrib_25.5.json @@ -0,0 +1,101 @@ +{ + "release_date": "2025-05-28", + "release_label": "25.5", + "release_product": "nvpl", + "nvpl_blas": { + "name": "NVPL BLAS", + "license": "NVPL", + "license_path": "nvpl_blas/LICENSE.txt", + "version": "0.4.1.1", + "linux-sbsa": { + "relative_path": "nvpl_blas/linux-sbsa/nvpl_blas-linux-sbsa-0.4.1.1-archive.tar.xz", + "sha256": "57704e2e211999c899bca26346b946b881b609554914245131b390410f7b93e8", + "md5": "9e2a95925dcb4bbe0ac337550f317272", + "size": "773716" + } + }, + "nvpl_common": { + "name": "NVPL Common", + "license": "NVPL", + "license_path": "nvpl_common/LICENSE.txt", + "version": "0.3.3", + "linux-sbsa": { + "relative_path": "nvpl_common/linux-sbsa/nvpl_common-linux-sbsa-0.3.3-archive.tar.xz", + "sha256": "fe87ccd63817427c6c9b9e292447a4e8f256b9c9157065fba1a338719fa433c8", + "md5": "9ae9f3253461a0565bc3c01a07c50fe3", + "size": "9444" + } + }, + "nvpl_fft": { + "name": "NVPL FFT", + "license": "NVPL", + "license_path": "nvpl_fft/LICENSE.txt", + "version": "0.4.2.1", + "linux-sbsa": { + "relative_path": "nvpl_fft/linux-sbsa/nvpl_fft-linux-sbsa-0.4.2.1-archive.tar.xz", + "sha256": "ebb9d98abc23ddee5c492e0bbf2c534570a38d7df1863a0630da2c6d7f5cca3d", + "md5": "211ec34eef6b023f7af8e1fc40ae4ad1", + "size": "24974464" + } + }, + "nvpl_lapack": { + "name": "NVPL LAPACK", + "license": "NVPL", + "license_path": "nvpl_lapack/LICENSE.txt", + "version": "0.3.1.1", + "linux-sbsa": { + "relative_path": "nvpl_lapack/linux-sbsa/nvpl_lapack-linux-sbsa-0.3.1.1-archive.tar.xz", + "sha256": "f5b916aa36a8549946fc2262acebb082fe8c463bd1523a3c0cc2c93527231653", + "md5": "ec0616b0be4616ac7050807a872343a2", + "size": "4781368" + } + }, + "nvpl_rand": { + "name": "NVPL RAND", + "license": "NVPL", + "license_path": "nvpl_rand/LICENSE.txt", + "version": "0.5.2", + "linux-sbsa": { + "relative_path": "nvpl_rand/linux-sbsa/nvpl_rand-linux-sbsa-0.5.2-archive.tar.xz", + "sha256": "1eb5c2a5e98390b2bc76c3218837916df64d33cce220169811e14ecead36933f", + "md5": "fe72dcf4600cbd85f7249e95ebbcd363", + "size": "34025772" + } + }, + "nvpl_scalapack": { + "name": "NVPL SCALAPACK", + "license": "NVPL", + "license_path": "nvpl_scalapack/LICENSE.txt", + "version": "0.2.2", + "linux-sbsa": { + "relative_path": "nvpl_scalapack/linux-sbsa/nvpl_scalapack-linux-sbsa-0.2.2-archive.tar.xz", + "sha256": "20cf6c54a0352f2fb0060e6f5ef6b892c5d07a242f8aab31cd9bbceb58a7bd11", + "md5": "8e3e728e8587cc4698beb2ab7ce162d6", + "size": "4647440" + } + }, + "nvpl_sparse": { + "name": "NVPL SPARSE", + "license": "NVPL", + "license_path": "nvpl_sparse/LICENSE.txt", + "version": "0.4.1", + "linux-sbsa": { + "relative_path": "nvpl_sparse/linux-sbsa/nvpl_sparse-linux-sbsa-0.4.1-archive.tar.xz", + "sha256": "fda868fe6619e94463a93efed1958e67588c607c170a4c658103a24295855e19", + "md5": "851080b3e56db9bf6fa37cea198bcb33", + "size": "556932" + } + }, + "nvpl_tensor": { + "name": "NVPL TENSOR", + "license": "NVPL", + "license_path": "nvpl_tensor/LICENSE.txt", + "version": "0.3.1", + "linux-sbsa": { + "relative_path": "nvpl_tensor/linux-sbsa/nvpl_tensor-linux-sbsa-0.3.1-archive.tar.xz", + "sha256": "12e9293609b3726cf9e92c648f117412a98a5e54700c877518ec2991e51ab50f", + "md5": "613b9a05d867667deae31bfea26688e8", + "size": "2338804" + } + } +} diff --git a/pkgs/development/cuda-modules/packages/nvpl_blas.nix b/pkgs/development/cuda-modules/packages/nvpl_blas.nix new file mode 100644 index 000000000000..94661a67067d --- /dev/null +++ b/pkgs/development/cuda-modules/packages/nvpl_blas.nix @@ -0,0 +1,18 @@ +{ buildRedist }: +buildRedist { + redistName = "cuda"; + pname = "nvpl_blas"; + + outputs = [ + "out" + "dev" + "include" + "lib" + ]; + + meta = { + description = "Part of NVIDIA Performance Libraries that provides standard Fortran 77 BLAS APIs as well as C (CBLAS)"; + homepage = "https://developer.nvidia.com/nvpl"; + changelog = "https://docs.nvidia.com/nvpl/latest/blas/release_notes.html"; + }; +} diff --git a/pkgs/development/cuda-modules/packages/nvpl_common.nix b/pkgs/development/cuda-modules/packages/nvpl_common.nix new file mode 100644 index 000000000000..86330f5187ed --- /dev/null +++ b/pkgs/development/cuda-modules/packages/nvpl_common.nix @@ -0,0 +1,16 @@ +{ buildRedist }: +buildRedist { + redistName = "cuda"; + pname = "nvpl_common"; + + outputs = [ + "out" + "dev" + ]; + + meta = { + description = "Common part of NVIDIA Performance Libraries"; + homepage = "https://developer.nvidia.com/nvpl"; + changelog = "https://docs.nvidia.com/nvpl/latest/release_notes.html"; + }; +} diff --git a/pkgs/development/cuda-modules/packages/nvpl_fft.nix b/pkgs/development/cuda-modules/packages/nvpl_fft.nix new file mode 100644 index 000000000000..30eb09ff9a11 --- /dev/null +++ b/pkgs/development/cuda-modules/packages/nvpl_fft.nix @@ -0,0 +1,18 @@ +{ buildRedist }: +buildRedist { + redistName = "cuda"; + pname = "nvpl_fft"; + + outputs = [ + "out" + "dev" + "include" + "lib" + ]; + + meta = { + description = "Perform Fast Fourier Transform (FFT) calculations on ARM CPUs"; + homepage = "https://developer.nvidia.com/nvpl"; + changelog = "https://docs.nvidia.com/nvpl/latest/fft/release_notes.html"; + }; +} diff --git a/pkgs/development/cuda-modules/packages/nvpl_lapack.nix b/pkgs/development/cuda-modules/packages/nvpl_lapack.nix new file mode 100644 index 000000000000..64637f45d9b5 --- /dev/null +++ b/pkgs/development/cuda-modules/packages/nvpl_lapack.nix @@ -0,0 +1,20 @@ +{ buildRedist, nvpl_blas }: +buildRedist { + redistName = "cuda"; + pname = "nvpl_lapack"; + + outputs = [ + "out" + "dev" + "include" + "lib" + ]; + + buildInputs = [ nvpl_blas ]; + + meta = { + description = "Part of NVIDIA Performance Libraries that provides standard Fortran 90 LAPACK and LAPACKE APIs"; + homepage = "https://developer.nvidia.com/nvpl"; + changelog = "https://docs.nvidia.com/nvpl/latest/lapack/release_notes.html"; + }; +} diff --git a/pkgs/development/cuda-modules/packages/nvpl_rand.nix b/pkgs/development/cuda-modules/packages/nvpl_rand.nix new file mode 100644 index 000000000000..f65e6f15f0ee --- /dev/null +++ b/pkgs/development/cuda-modules/packages/nvpl_rand.nix @@ -0,0 +1,18 @@ +{ buildRedist }: +buildRedist { + redistName = "cuda"; + pname = "nvpl_rand"; + + outputs = [ + "out" + "dev" + "include" + "lib" + ]; + + meta = { + description = "Collection of efficient pseudorandom and quasirandom number generators for ARM CPUs"; + homepage = "https://developer.nvidia.com/nvpl"; + changelog = "https://docs.nvidia.com/nvpl/latest/rand/release_notes.html"; + }; +} diff --git a/pkgs/development/cuda-modules/packages/nvpl_scalapack.nix b/pkgs/development/cuda-modules/packages/nvpl_scalapack.nix new file mode 100644 index 000000000000..d9bf55082b6c --- /dev/null +++ b/pkgs/development/cuda-modules/packages/nvpl_scalapack.nix @@ -0,0 +1,18 @@ +{ buildRedist }: +buildRedist { + redistName = "cuda"; + pname = "nvpl_scalapack"; + + outputs = [ + "out" + "dev" + "include" + "lib" + ]; + + meta = { + description = "Provides an optimized implementation of ScaLAPACK for distributed-memory architectures"; + homepage = "https://developer.nvidia.com/nvpl"; + changelog = "https://docs.nvidia.com/nvpl/latest/scalapack/release_notes.html"; + }; +} diff --git a/pkgs/development/cuda-modules/packages/nvpl_sparse.nix b/pkgs/development/cuda-modules/packages/nvpl_sparse.nix new file mode 100644 index 000000000000..3bd6985e66e0 --- /dev/null +++ b/pkgs/development/cuda-modules/packages/nvpl_sparse.nix @@ -0,0 +1,19 @@ +{ buildRedist }: +buildRedist { + redistName = "cuda"; + pname = "nvpl_sparse"; + + outputs = [ + "out" + "dev" + "include" + "lib" + "static" + ]; + + meta = { + description = "Provides a set of CPU-accelerated basic linear algebra subroutines used for handling sparse matrices"; + homepage = "https://developer.nvidia.com/nvpl"; + changelog = "https://docs.nvidia.com/nvpl/latest/sparse/release_notes.html"; + }; +} diff --git a/pkgs/development/cuda-modules/packages/nvpl_tensor.nix b/pkgs/development/cuda-modules/packages/nvpl_tensor.nix new file mode 100644 index 000000000000..177c267465bd --- /dev/null +++ b/pkgs/development/cuda-modules/packages/nvpl_tensor.nix @@ -0,0 +1,20 @@ +{ buildRedist, nvpl_blas }: +buildRedist { + redistName = "cuda"; + pname = "nvpl_tensor"; + + outputs = [ + "out" + "dev" + "include" + "lib" + ]; + + buildInputs = [ nvpl_blas ]; + + meta = { + description = "Part of NVIDIA Performance Libraries that provides tensor primitives"; + homepage = "https://developer.nvidia.com/nvpl"; + changelog = "https://docs.nvidia.com/nvpl/latest/tensor/release_notes.html"; + }; +} diff --git a/pkgs/top-level/cuda-packages.nix b/pkgs/top-level/cuda-packages.nix index 7c18919d703e..e6a4a3355e45 100644 --- a/pkgs/top-level/cuda-packages.nix +++ b/pkgs/top-level/cuda-packages.nix @@ -30,6 +30,7 @@ in cudnn = "9.13.0"; cusparselt = "0.6.3"; cutensor = "2.3.1"; + nvpl = "25.5"; tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.9.0"; }; }; @@ -41,6 +42,7 @@ in cudnn = "9.13.0"; cusparselt = "0.8.1"; cutensor = "2.3.1"; + nvpl = "25.5"; tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.9.0"; }; }; @@ -52,6 +54,7 @@ in cudnn = "9.13.0"; cusparselt = "0.8.1"; cutensor = "2.3.1"; + nvpl = "25.5"; tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.9.0"; }; }; @@ -63,6 +66,7 @@ in cudnn = "9.13.0"; cusparselt = "0.8.1"; cutensor = "2.3.1"; + nvpl = "25.5"; tensorrt = if hasJetsonCudaCapability then "10.7.0" else "10.9.0"; }; };