2024-02-25 17:40:43 +08:00
|
|
|
{
|
|
|
|
buildFHSEnv, writeScript, stdenvNoCC, requireFile, substituteAll, symlinkJoin,
|
2024-03-16 14:47:52 +08:00
|
|
|
config, oneapiArch ? config.oneapiArch or "SSE3", additionalCommands ? "",
|
2024-03-12 18:44:22 +08:00
|
|
|
oneapi, gcc, glibc, lmod, rsync, which, wannier90, binutils, hdf5
|
2024-02-25 17:40:43 +08:00
|
|
|
}:
|
|
|
|
let
|
2024-03-12 15:16:54 +08:00
|
|
|
sources = import ../source.nix { inherit requireFile; };
|
2024-02-25 17:40:43 +08:00
|
|
|
buildEnv = buildFHSEnv
|
|
|
|
{
|
|
|
|
name = "buildEnv";
|
|
|
|
# make "module load mpi" success
|
2024-03-12 15:16:54 +08:00
|
|
|
targetPkgs = pkgs: with pkgs; [ zlib (writeTextDir "etc/release" "") gccFull ];
|
2024-02-25 17:40:43 +08:00
|
|
|
};
|
|
|
|
buildScript = writeScript "build"
|
|
|
|
''
|
|
|
|
. ${lmod}/share/lmod/lmod/init/bash
|
|
|
|
module use ${oneapi}/share/intel/modulefiles
|
|
|
|
module load tbb compiler-rt oclfpga # dependencies
|
|
|
|
module load mpi mkl compiler
|
|
|
|
mkdir -p bin
|
2024-03-12 15:16:54 +08:00
|
|
|
make DEPS=1 -j$NIX_BUILD_CORES
|
2024-02-25 17:40:43 +08:00
|
|
|
'';
|
|
|
|
include = version: substituteAll
|
|
|
|
{
|
|
|
|
src = ./makefile.include-${version};
|
|
|
|
inherit oneapiArch;
|
|
|
|
};
|
2024-03-12 15:16:54 +08:00
|
|
|
gccFull = symlinkJoin { name = "gcc"; paths = [ gcc gcc.cc gcc.cc.lib glibc.dev binutils.bintools ]; };
|
2024-02-25 17:40:43 +08:00
|
|
|
vasp = version: stdenvNoCC.mkDerivation rec
|
|
|
|
{
|
2024-03-12 18:44:22 +08:00
|
|
|
pname = "vasp-intel";
|
2024-02-25 17:40:43 +08:00
|
|
|
inherit version;
|
2024-03-12 15:16:54 +08:00
|
|
|
src = sources.${version};
|
2024-02-25 17:40:43 +08:00
|
|
|
configurePhase =
|
|
|
|
''
|
|
|
|
cp ${include version} makefile.include
|
|
|
|
cp ${../constr_cell_relax.F} src/constr_cell_relax.F
|
|
|
|
'';
|
2024-03-12 15:16:54 +08:00
|
|
|
nativeBuildInputs = [ rsync which ];
|
2024-03-12 18:44:22 +08:00
|
|
|
HDF5_ROOT = hdf5;
|
2024-02-25 17:40:43 +08:00
|
|
|
WANNIER90_ROOT = wannier90;
|
|
|
|
buildPhase = "${buildEnv}/bin/buildEnv ${buildScript}";
|
|
|
|
installPhase =
|
|
|
|
''
|
|
|
|
mkdir -p $out/bin
|
|
|
|
for i in std gam ncl; do cp bin/vasp_$i $out/bin/vasp-$i; done
|
|
|
|
'';
|
2024-03-13 12:05:00 +08:00
|
|
|
dontFixup = true;
|
2024-03-12 18:44:22 +08:00
|
|
|
requiredSystemFeatures = [ "gccarch-exact-${stdenvNoCC.hostPlatform.gcc.arch}" "big-parallel" ];
|
2024-02-25 17:40:43 +08:00
|
|
|
};
|
|
|
|
startScript = version: writeScript "vasp-intel-${version}"
|
|
|
|
''
|
|
|
|
. ${lmod}/share/lmod/lmod/init/bash
|
|
|
|
module use ${oneapi}/share/intel/modulefiles
|
|
|
|
module load tbb compiler-rt oclfpga # dependencies
|
|
|
|
module load mpi mkl compiler
|
2024-03-12 23:57:59 +08:00
|
|
|
|
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-03-12 23:57:59 +08:00
|
|
|
fi
|
2024-03-13 12:05:00 +08:00
|
|
|
|
2024-03-16 14:47:52 +08:00
|
|
|
${additionalCommands}
|
|
|
|
|
2024-02-26 19:34:40 +08:00
|
|
|
exec "$@"
|
2024-02-25 17:40:43 +08:00
|
|
|
'';
|
|
|
|
runEnv = version: buildFHSEnv
|
|
|
|
{
|
|
|
|
name = "vasp-intel-${version}";
|
2024-03-13 12:05:00 +08:00
|
|
|
targetPkgs = pkgs: with pkgs; [ zlib (vasp version) (writeTextDir "etc/release" "") gccFull ];
|
2024-02-25 17:40:43 +08:00
|
|
|
runScript = startScript version;
|
2024-03-13 15:58:20 +08:00
|
|
|
extraInstallCommands =
|
|
|
|
"for i in std gam ncl; do ln -s ${vasp version}/bin/vasp-$i $out/bin/vasp-intel-${version}-$i; done";
|
2024-02-25 17:40:43 +08:00
|
|
|
};
|
2024-03-12 15:16:54 +08:00
|
|
|
in builtins.mapAttrs (version: _: runEnv version) sources
|