armpl: add new package (#30305)

Co-authored-by: Annop Wongwathanarat <annop.wongwathanarat@arm.com>

1 files changed, 179 insertions, 0 deletions

diff --git a/var/spack/repos/builtin/packages/armpl/package.py b/var/spack/repos/builtin/packages/armpl/package.py
new file mode 100644
index 0000000000..4928404ba0
--- /dev/null
+++ b/var/spack/repos/builtin/packages/armpl/package.py
@@ -0,0 +1,179 @@
+# Copyright 2013-2022 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+import os
+
+from spack import *
+
+_os_map = {
+    'ubuntu18.04': 'Ubuntu-18.04',
+    'ubuntu20.04': 'Ubuntu-20.04',
+    'sles15': 'SLES-15',
+    'centos7': 'RHEL-7',
+    'centos8': 'RHEL-8',
+    'amzn2': 'RHEL-7'
+}
+
+
+_versions = {
+    '22.0.1_gcc-11.2': {
+        'RHEL-7': ('32529fdc70c39084eafe746db6baa487815bb49dae1588ccf2f7bd7929c10f7c'),
+        'RHEL-8': ('1abd0b1c47cae65ee74510cf6e25946c66f9f4244e4674d5e9f73c442901482c'),
+        'SLES-15': ('631261d7b29e85e99d208bdd397bdb5fcb0b53fd834b4b5a9cf963789e29f96e'),
+        'Ubuntu-18.04': ('2ec210ff3c33d94a7cbe0cd7ea92add0b954ab1b1dc438dffa1d5f516e80e3ec'),
+        'Ubuntu-20.04': ('13e9b98afc01c5444799c7f2ef84716f8a7be11df93b71fe9cfdc7fc6162f6d5')
+    },
+    '22.0.1_gcc-10.2': {
+        'RHEL-7': ('d2d91f43872e072ccec0cfab61eccf531daf6f02997e29fef3d738178c023d7a'),
+        'RHEL-8': ('d642f55937410d2d402589f09e985c05b577d1227063b8247dc5733199e124a4'),
+        'SLES-15': ('6746de2db361a65edac2ff8dcd4fc84a314fd919df3758c9bad7027dcfadfea2'),
+        'Ubuntu-18.04': ('f3a7a7cb1768046ef742110fa311c65504074f1a381d295d583848221e267bd9'),
+        'Ubuntu-20.04': ('5da7450196d94b0aea613cf8e7c4083ae3eb2e905d049db3b300059a9fbf169b')
+    },
+    '22.0.1_gcc-9.3': {
+        'RHEL-7': ('8df55f83ccebf9c1de5291c701d7cbeb051ce194ffe2d1f1148b2a6be0d7ea1c'),
+        'RHEL-8': ('b0e26004c40db3138939b7bddc4bbe54ec7de4e548b5dc697cce5c85a8acbb27'),
+        'SLES-15': ('963278d35485ec28a8b17a89efcfe0f82d84edc4ff8af838d56648917ec7b547'),
+        'Ubuntu-18.04': ('3c7d2f7d102954440539d5b541dd1f669d2ccb3daaa14de1f04d6790368d6794'),
+        'Ubuntu-20.04': ('8e78bef6517f42efd878579aee2cae4e439e3cd5c8a28e3f3fa83254f7189a2f')
+    },
+    '22.0.1_gcc-8.2': {
+        'RHEL-7': ('1e682e319c3b07236acc3870bf291a1d0cba884112b447dad7e356fdc42bd06a'),
+        'RHEL-8': ('1fad5a0de02cda0a23a7864cca653a04ceb4244e362073d2959ce7db4144bb20'),
+        'SLES-15': ('fa6111264c3fbe29ec084e7322c794640a1b3c40b2f0e01f7637f3f0d87d03e2'),
+        'Ubuntu-18.04': ('3d092ecd98620b31e813ad726244ff40fdcb012aa055b6695dff51bc1578039d')
+    },
+    '22.0.1_gcc-7.5': {
+        'RHEL-7': ('e23702a9fecfc64aa6bd56439a602f0c25b0febce059cb6c0192b575758c6f1a'),
+        'Ubuntu-18.04': ('bf4e6327eedec656b696f98735aa988a75b0c60185f3c22af6b7e608abbdb305')
+    }
+}
+
+
+def get_os():
+    spack_os = spack.platforms.host().default_os
+    return _os_map.get(spack_os, 'RHEL-7')
+
+
+def get_package_url(version):
+    os = get_os()
+    os_no_dash = get_os().replace('-', '')
+    base_url = 'https://developer.arm.com/-/media/Files/downloads/hpc/arm-performance-libraries/'
+    armpl_version = version.split('_')[0]
+    armpl_version_dashed = armpl_version.replace('.', '-')
+    gcc_version = version.split('_')[1]
+    filename = 'arm-performance-libraries_' + armpl_version + '_' + os + '_'\
+        + gcc_version + '.tar'
+    return (base_url + armpl_version_dashed + '/' + os_no_dash + '/' + filename)
+
+
+def get_armpl_prefix(spec):
+    return os.path.join(spec.prefix, 'armpl_' + spec.version.string)
+
+
+class Armpl(Package):
+    """Arm Performance Libraries provides optimized standard core math libraries for
+    high-performance computing applications on Arm processors."""
+
+    homepage = "https://developer.arm.com/tools-and-software/server-and-hpc/downloads/arm-performance-libraries"
+    url = "https://developer.arm.com/-/media/Files/downloads/hpc/arm-performance-libraries/22-0-1/RHEL7/arm-performance-libraries_22.0.1_RHEL-7_gcc-11.2.tar"
+
+    maintainers = ['annop-w']
+
+    for ver, packages in _versions.items():
+        key = "{0}".format(get_os())
+        sha256sum = packages.get(key)
+        url = get_package_url(ver)
+        if sha256sum:
+            version(ver, sha256=sha256sum, url=url)
+
+    conflicts('target=x86:', msg='Only available on Aarch64')
+    conflicts('target=ppc64:', msg='Only available on Aarch64')
+    conflicts('target=ppc64le:', msg='Only available on Aarch64')
+
+    conflicts('%gcc@:11.0', when='@22.0.1_gcc-11.2')
+    conflicts('%gcc@:10.0', when='@22.0.1_gcc-10.2')
+    conflicts('%gcc@:9.0', when='@22.0.1_gcc-9.3')
+    conflicts('%gcc@:8.0', when='@22.0.1_gcc-8.2')
+    conflicts('%gcc@:7.0', when='@22.0.1_gcc-7.5')
+
+    variant('ilp64', default=False, description='use ilp64 specific Armpl library')
+    variant('shared', default=True, description='enable shared libs')
+    variant(
+        'threads', default='none',
+        description='Multithreading support',
+        values=('openmp', 'none'),
+        multi=False
+    )
+
+    provides('blas')
+    provides('lapack')
+    provides('fftw-api@3')
+
+    # Run the installer with the desired install directory
+    def install(self, spec, prefix):
+        if self.compiler.name != 'gcc':
+            raise spack.error.SpackError(('Only compatible with GCC.\n'))
+
+        exe = Executable('./arm-performance-libraries_{0}_{1}.sh'.format(
+            spec.version.up_to(3), get_os())
+        )
+        exe("--accept", "--force", "--install-to", prefix)
+
+    @property
+    def blas_libs(self):
+
+        armpl_prefix = get_armpl_prefix(self.spec)
+
+        shared = True if '+shared' in self.spec else False
+        if '+ilp64' in self.spec and self.spec.satisfies('threads=openmp'):
+            libname = 'libarmpl_ilp64_mp'
+        elif '+ilp64' in self.spec:
+            libname = 'libarmpl_ilp64'
+        elif self.spec.satisfies('threads=openmp'):
+            libname = 'libarmpl_mp'
+        else:
+            libname = 'libarmpl'
+
+        # Get ArmPL Lib
+        armpl_libs = find_libraries(
+            [libname, 'libamath', 'libastring'],
+            root=armpl_prefix,
+            shared=shared,
+            recursive=True)
+
+        armpl_libs += find_system_libraries(['libm'])
+
+        return armpl_libs
+
+    @property
+    def lapack_libs(self):
+        return self.blas_libs
+
+    @property
+    def fftw_libs(self):
+        return self.blas_libs
+
+    @property
+    def libs(self):
+        return self.blas_libs
+
+    @property
+    def headers(self):
+        armpl_dir = get_armpl_prefix(self.spec)
+        suffix = 'include'
+        if self.spec.satisfies('+ilp64'):
+            suffix += '_ilp64'
+        if self.spec.satisfies('threads=openmp'):
+            suffix += '_mp'
+        incdir = join_path(armpl_dir, suffix)
+
+        hlist = find_all_headers(incdir)
+        hlist.directories = [incdir]
+        return hlist
+
+    def setup_run_environment(self, env):
+        armpl_dir = get_armpl_prefix(self.spec)
+        env.prepend_path("LD_LIBRARY_PATH", join_path(armpl_dir, "lib"))