diff options
author | Annop Wongwathanarat <annop.wongwathanarat@gmail.com> | 2023-03-08 09:25:36 +0000 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-03-08 10:25:36 +0100 |
commit | 983a56e7295479cdd0ec13695a629884f098ccc4 (patch) | |
tree | 92ef525d9d3f92c40806f81014da0e054bd7ecc3 /var | |
parent | cbd0770497eed68fd348dce29910e8a7fdff74ae (diff) | |
download | spack-983a56e7295479cdd0ec13695a629884f098ccc4.tar.gz spack-983a56e7295479cdd0ec13695a629884f098ccc4.tar.bz2 spack-983a56e7295479cdd0ec13695a629884f098ccc4.tar.xz spack-983a56e7295479cdd0ec13695a629884f098ccc4.zip |
gromacs: add sve variant on aarch64 (#35614)
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": |