diff options
Diffstat (limited to 'var')
-rw-r--r-- | var/spack/repos/builtin/packages/gromacs/package.py | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/gromacs/package.py b/var/spack/repos/builtin/packages/gromacs/package.py index 7804b7ef20..8a28c76687 100644 --- a/var/spack/repos/builtin/packages/gromacs/package.py +++ b/var/spack/repos/builtin/packages/gromacs/package.py @@ -114,6 +114,15 @@ class Gromacs(CMakePackage): ) variant("openmp", default=True, description="Enables OpenMP at configure time") variant( + "sve", + default=True, + description="Enable SVE on aarch64 if available", + when="target=neoverse_v1", + ) + variant( + "sve", default=True, description="Enable SVE on aarch64 if available", when="target=a64fx" + ) + variant( "relaxed_double_precision", default=False, description="GMX_RELAXED_DOUBLE_PRECISION, use only for Fujitsu PRIMEHPC", @@ -450,6 +459,8 @@ class Gromacs(CMakePackage): # ARMv8 if self.spec.satisfies("%nvhpc"): options.append("-DGMX_SIMD=None") + elif "sve" in target.features and "+sve" in self.spec: + options.append("-DGMX_SIMD=ARM_SVE") else: options.append("-DGMX_SIMD=ARM_NEON_ASIMD") elif target == "mic_knl": |