forked from mirrors/nixpkgs
58e1666a58
* make CPU acceleration selectable. * update cmake flag for acceleration CPU_ACCELERATION -> SIMD GMX_CPU_ACCELERATION is outdated and has been ignored. * add hwloc to inputs * always build with OpenMP * change name -> pname
83 lines
2.5 KiB
Nix
83 lines
2.5 KiB
Nix
{ stdenv
|
|
, fetchurl
|
|
, cmake
|
|
, hwloc
|
|
, fftw
|
|
, openmpi
|
|
, perl
|
|
, singlePrec ? true
|
|
, mpiEnabled ? false
|
|
, cpuAcceleration ? null
|
|
}:
|
|
|
|
let
|
|
# Select reasonable defaults for all major platforms
|
|
# The possible values are defined in CMakeLists.txt:
|
|
# AUTO None SSE2 SSE4.1 AVX_128_FMA AVX_256 AVX2_256
|
|
# AVX2_128 AVX_512 AVX_512_KNL MIC ARM_NEON ARM_NEON_ASIMD
|
|
SIMD = x: if (cpuAcceleration != null) then x else
|
|
if stdenv.hostPlatform.system == "i686-linux" then "SSE2" else
|
|
if stdenv.hostPlatform.system == "x86_64-linux" then "SSE4.1" else
|
|
if stdenv.hostPlatform.system == "x86_64-darwin" then "SSE4.1" else
|
|
if stdenv.hostPlatform.system == "aarch64-linux" then "ARM_NEON" else
|
|
"None";
|
|
|
|
in stdenv.mkDerivation rec {
|
|
pname = "gromacs";
|
|
version = "2020.4";
|
|
|
|
src = fetchurl {
|
|
url = "ftp://ftp.gromacs.org/pub/gromacs/gromacs-${version}.tar.gz";
|
|
sha256 = "1rplvgna60nqyb8nspaz3bfkwb044kv3zxdaa5whql5m441nj6am";
|
|
};
|
|
|
|
nativeBuildInputs = [ cmake ];
|
|
buildInputs = [ fftw perl hwloc ]
|
|
++ (stdenv.lib.optionals mpiEnabled [ openmpi ]);
|
|
|
|
cmakeFlags = [
|
|
"-DGMX_SIMD:STRING=${SIMD cpuAcceleration}"
|
|
"-DGMX_OPENMP:BOOL=TRUE"
|
|
] ++ (
|
|
if singlePrec then [
|
|
"-DGMX_DOUBLE=OFF"
|
|
] else [
|
|
"-DGMX_DOUBLE=ON"
|
|
"-DGMX_DEFAULT_SUFFIX=OFF"
|
|
]
|
|
) ++ (
|
|
if mpiEnabled then [
|
|
"-DGMX_MPI:BOOL=TRUE"
|
|
"-DGMX_THREAD_MPI:BOOL=FALSE"
|
|
] else [
|
|
"-DGMX_MPI:BOOL=FALSE"
|
|
]
|
|
);
|
|
|
|
meta = with stdenv.lib; {
|
|
homepage = "http://www.gromacs.org";
|
|
license = licenses.gpl2;
|
|
description = "Molecular dynamics software package";
|
|
longDescription = ''
|
|
GROMACS is a versatile package to perform molecular dynamics,
|
|
i.e. simulate the Newtonian equations of motion for systems
|
|
with hundreds to millions of particles.
|
|
|
|
It is primarily designed for biochemical molecules like
|
|
proteins, lipids and nucleic acids that have a lot of
|
|
complicated bonded interactions, but since GROMACS is
|
|
extremely fast at calculating the nonbonded interactions (that
|
|
usually dominate simulations) many groups are also using it
|
|
for research on non-biological systems, e.g. polymers.
|
|
|
|
GROMACS supports all the usual algorithms you expect from a
|
|
modern molecular dynamics implementation, (check the online
|
|
reference or manual for details), but there are also quite a
|
|
few features that make it stand out from the competition.
|
|
|
|
See: http://www.gromacs.org/About_Gromacs for details.
|
|
'';
|
|
platforms = platforms.unix;
|
|
};
|
|
}
|