2023-09-03 22:46:18 +08:00
|
|
|
{
|
2024-03-18 19:52:25 +08:00
|
|
|
stdenvNoCC, requireFile, writeShellApplication,
|
2024-03-16 14:47:52 +08:00
|
|
|
rsync, blas, scalapack, mpi, openmp, gfortran, gcc, fftwMpi, hdf5, wannier90,
|
|
|
|
additionalCommands ? ""
|
2023-09-03 22:46:18 +08:00
|
|
|
}:
|
2024-02-24 22:39:48 +08:00
|
|
|
let
|
2024-02-27 12:47:15 +08:00
|
|
|
sources = import ../source.nix { inherit requireFile; };
|
2024-03-10 14:01:32 +08:00
|
|
|
include = version: ./makefile.include-${version};
|
2024-02-24 22:39:48 +08:00
|
|
|
vasp = version: stdenvNoCC.mkDerivation rec
|
2023-09-03 22:46:18 +08:00
|
|
|
{
|
2024-02-24 22:42:42 +08:00
|
|
|
pname = "vasp-gnu";
|
2024-02-24 22:39:48 +08:00
|
|
|
inherit version;
|
2024-02-27 12:47:15 +08:00
|
|
|
src = sources.${version};
|
2024-02-24 22:39:48 +08:00
|
|
|
configurePhase =
|
|
|
|
''
|
2024-03-09 15:12:45 +08:00
|
|
|
cp ${include version} makefile.include
|
2024-02-25 15:34:35 +08:00
|
|
|
cp ${../constr_cell_relax.F} src/constr_cell_relax.F
|
2024-02-24 22:39:48 +08:00
|
|
|
mkdir -p bin
|
|
|
|
'';
|
|
|
|
enableParallelBuilding = true;
|
|
|
|
makeFlags = "DEPS=1";
|
2024-02-25 15:34:35 +08:00
|
|
|
buildInputs = [ blas scalapack mpi openmp fftwMpi.dev fftwMpi hdf5 hdf5.dev wannier90 ];
|
2024-02-24 22:39:48 +08:00
|
|
|
nativeBuildInputs = [ rsync gfortran gfortran.cc gcc ];
|
|
|
|
FFTW_ROOT = fftwMpi.dev;
|
2024-02-25 15:34:35 +08:00
|
|
|
HDF5_ROOT = hdf5.dev;
|
2024-03-10 14:01:32 +08:00
|
|
|
WANNIER90_ROOT = wannier90;
|
2024-02-24 22:39:48 +08:00
|
|
|
installPhase =
|
|
|
|
''
|
|
|
|
mkdir -p $out/bin
|
|
|
|
for i in std gam ncl; do
|
2024-02-26 19:34:40 +08:00
|
|
|
cp bin/vasp_$i $out/bin/vasp-$i
|
2024-02-24 22:39:48 +08:00
|
|
|
done
|
|
|
|
'';
|
2023-09-03 22:46:18 +08:00
|
|
|
};
|
2024-02-26 19:34:40 +08:00
|
|
|
startScript = version: writeShellApplication
|
|
|
|
{
|
|
|
|
name = "vasp-gnu-${version}";
|
|
|
|
runtimeInputs = [ (vasp version) ];
|
|
|
|
text =
|
|
|
|
''
|
2024-03-13 17:21:02 +08:00
|
|
|
# if SLURM_CPUS_PER_TASK is set, use it to set OMP_NUM_THREADS
|
|
|
|
if [ -n "''${SLURM_CPUS_PER_TASK-}" ]; then
|
|
|
|
export OMP_NUM_THREADS=$SLURM_CPUS_PER_TASK
|
2024-02-26 19:34:40 +08:00
|
|
|
fi
|
2024-03-16 14:47:52 +08:00
|
|
|
|
|
|
|
${additionalCommands}
|
|
|
|
|
2024-02-26 19:34:40 +08:00
|
|
|
exec "$@"
|
|
|
|
'';
|
|
|
|
};
|
2024-02-27 12:47:15 +08:00
|
|
|
in builtins.mapAttrs (version: _: startScript version) sources
|