diff options
author | Phillip Seeber <phillip.seeber@uni-jena.de> | 2023-10-05 17:29:12 +0200 |
---|---|---|
committer | Phillip Seeber <phillip.seeber@uni-jena.de> | 2023-10-09 10:48:16 +0200 |
commit | 5b7e0ad0962ca3e50879d008f0ea98ead9ee051f (patch) | |
tree | aac67a71158740842edd723a502db9a4a60216e2 /pkgs/by-name/sp | |
parent | 51d261b0aed53b56dd4a658a05c9ed6b937eb731 (diff) |
spla: init at 1.0.6
spla: fix openmp on darwin
Diffstat (limited to 'pkgs/by-name/sp')
-rw-r--r-- | pkgs/by-name/sp/spla/package.nix | 66 |
1 files changed, 66 insertions, 0 deletions
diff --git a/pkgs/by-name/sp/spla/package.nix b/pkgs/by-name/sp/spla/package.nix new file mode 100644 index 0000000000000..3143fbeb73165 --- /dev/null +++ b/pkgs/by-name/sp/spla/package.nix @@ -0,0 +1,66 @@ +{ stdenv +, lib +, fetchFromGitHub +, cmake +, mpi +, blas +, gfortran +, llvmPackages +, gpuBackend ? "none" +, cudaPackages +, hip +, rocblas +}: + +assert builtins.elem gpuBackend [ "none" "cuda" "rocm" ]; + +stdenv.mkDerivation rec { + pname = "spla"; + version = "1.5.5"; + + src = fetchFromGitHub { + owner = "eth-cscs"; + repo = pname; + rev = "v${version}"; + hash = "sha256-71QpwTsRogH+6Bik9DKwezl9SqwoLxQt4SZ7zw5X6DE="; + }; + + postPatch = '' + substituteInPlace src/gpu_util/gpu_blas_api.hpp \ + --replace '#include <rocblas.h>' '#include <rocblas/rocblas.h>' + ''; + + nativeBuildInputs = [ + cmake + gfortran + ]; + + buildInputs = [ + blas + ] + ++ lib.optional (gpuBackend == "cuda") cudaPackages.cudatoolkit + ++ lib.optionals (gpuBackend == "rocm") [ hip rocblas rocblas ] + ++ lib.optional stdenv.isDarwin llvmPackages.openmp + ; + + propagatedBuildInputs = [ mpi ]; + + cmakeFlags = [ + "-DSPLA_OMP=ON" + "-DSPLA_FORTRAN=ON" + "-DSPLA_INSTALL=ON" + # Required due to broken CMake files + "-DCMAKE_INSTALL_LIBDIR=lib" + "-DCMAKE_INSTALL_INCLUDEDIR=include" + ] + ++ lib.optional (gpuBackend == "cuda") "-DSPLA_GPU_BACKEND=CUDA" + ++ lib.optional (gpuBackend == "rocm") [ "-DSPLA_GPU_BACKEND=ROCM" ] + ; + + 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 ];# + }; +} |