OpenBLAS: Allow enabling/disabling AVX2/512 support

Allow specifying whether AVX2 or AVX512 should be enabled or disabled
on x86 targets.

As AVX2 hardware and toolchain support is quite ubiquitous by now,
AVX2 is enabled by default.  Also AVX2 support is not disabled when
building the +virtual_machine variant.

AVX512 is not supported in older but still supported toolchains,
hardware is still expensive, and OpenBLAS AVX512 kernels still have
bugs. Thus AVX512 is disabled by default.

1 files changed, 18 insertions, 1 deletions

diff --git a/var/spack/repos/builtin/packages/openblas/package.py b/var/spack/repos/builtin/packages/openblas/package.py
index b3a64a6af7..df9145b29a 100644
--- a/var/spack/repos/builtin/packages/openblas/package.py
+++ b/var/spack/repos/builtin/packages/openblas/package.py
@@ -56,6 +56,18 @@ class Openblas(MakefilePackage):
         description="Adding options to build openblas on Linux virtual machine"
     )
 
+    variant(
+        'avx2',
+        default=True,
+        description='Enable use of AVX2 instructions'
+    )
+
+    variant(
+        'avx512',
+        default=False,
+        description='Enable use of AVX512 instructions'
+    )
+
     # virtual dependency
     provides('blas')
     provides('lapack')
@@ -147,7 +159,6 @@ class Openblas(MakefilePackage):
         if self.spec.variants['virtual_machine'].value:
             make_defs += [
                 'DYNAMIC_ARCH=1',
-                'NO_AVX2=1',
                 'NUM_THREADS=64',  # OpenBLAS stores present no of CPUs as max
             ]
 
@@ -185,6 +196,12 @@ class Openblas(MakefilePackage):
         if '+ilp64' in self.spec:
             make_defs += ['INTERFACE64=1']
 
+        if 'x86' in spack.architecture.sys_type():
+            if '~avx2' in self.spec:
+                make_defs += ['NO_AVX2=1']
+            if '~avx512' in self.spec:
+                make_defs += ['NO_AVX512=1']
+
         return make_defs
 
     @property