nixos/local/pkgs/vasp-gpu/default.nix
2024-02-21 15:27:39 +08:00

43 lines
1.1 KiB
Nix

{ buildFHSEnv, writeScript, stdenvNoCC, requireFile, nvhpc, lmod }:
let
env = buildFHSEnv
{
name = "env";
targetPkgs = pkgs: with pkgs; [ nvhpc gfortran zlib which rsync mkl lmod ];
};
buildScript = writeScript "build"
''
. /usr/share/lmod/lmod/init/bash
module use /usr/share/nvhpc/modulefiles
module load nvhpc
mkdir -p bin
export MKLROOT=/usr
make DEPS=1 -j$NIX_BUILD_CORES
'';
vasp = stdenvNoCC.mkDerivation rec
{
pname = "vasp";
version = "6.4.0";
# nix-store --query --hash $(nix store add-path ./vasp-6.4.0)
src = requireFile
{
name = "${pname}-${version}";
sha256 = "189i1l5q33ynmps93p2mwqf5fx7p4l50sls1krqlv8ls14s3m71f";
hashMode = "recursive";
message = "Source file not found.";
};
configurePhase = "cp ${./makefile.include} makefile.include";
enableParallelBuilding = true;
buildPhase = "${env}/bin/env ${buildScript}";
installPhase =
''
mkdir -p $out
cp -r bin $out
'';
};
in buildFHSEnv
{
name = "vasp-gpu";
targetPkgs = pkgs: with pkgs; [ nvhpc gfortran zlib mkl lmod vasp ];
}