diff options
Diffstat (limited to 'pkgs/by-name/sp/spla')
-rw-r--r-- | pkgs/by-name/sp/spla/package.nix | 18 |
1 files changed, 12 insertions, 6 deletions
diff --git a/pkgs/by-name/sp/spla/package.nix b/pkgs/by-name/sp/spla/package.nix index 210fb04af9b80..2e07da1cdb020 100644 --- a/pkgs/by-name/sp/spla/package.nix +++ b/pkgs/by-name/sp/spla/package.nix @@ -22,15 +22,17 @@ assert builtins.elem gpuBackend [ "none" "cuda" "rocm" ]; stdenv.mkDerivation rec { pname = "spla"; - version = "1.5.5"; + version = "1.6.1"; src = fetchFromGitHub { owner = "eth-cscs"; repo = pname; rev = "v${version}"; - hash = "sha256-71QpwTsRogH+6Bik9DKwezl9SqwoLxQt4SZ7zw5X6DE="; + hash = "sha256-fNH1IOKV1Re8G7GH9Xfn3itR80eonTbEGKQRRD16/2k="; }; + outputs = [ "out" "dev" ]; + postPatch = '' substituteInPlace src/gpu_util/gpu_blas_api.hpp \ --replace '#include <rocblas.h>' '#include <rocblas/rocblas.h>' @@ -43,16 +45,15 @@ stdenv.mkDerivation rec { buildInputs = [ blas + mpi ] ++ lib.optional (gpuBackend == "cuda") cudaPackages.cudatoolkit ++ lib.optionals (gpuBackend == "rocm") [ rocmPackages.clr rocmPackages.rocblas - ] ++ lib.optional stdenv.isDarwin llvmPackages.openmp + ] ++ lib.optional stdenv.hostPlatform.isDarwin llvmPackages.openmp ; - propagatedBuildInputs = [ mpi ]; - cmakeFlags = [ "-DSPLA_OMP=ON" "-DSPLA_FORTRAN=ON" @@ -65,10 +66,15 @@ stdenv.mkDerivation rec { ++ lib.optional (gpuBackend == "rocm") [ "-DSPLA_GPU_BACKEND=ROCM" ] ; + preFixup = '' + substituteInPlace $out/lib/cmake/SPLA/SPLASharedTargets-release.cmake \ + --replace-fail "\''${_IMPORT_PREFIX}" "$out" + ''; + meta = with lib; { description = "Specialized Parallel Linear Algebra, providing distributed GEMM functionality for specific matrix distributions with optional GPU acceleration"; homepage = "https://github.com/eth-cscs/spla"; license = licenses.bsd3; - maintainers = [ maintainers.sheepforce ];# + maintainers = [ maintainers.sheepforce ]; }; } |