288 lines
11 KiB
Python
288 lines
11 KiB
Python
# Copyright 2013-2023 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 re
|
|
|
|
from spack.package 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.1": {
|
|
"RHEL-7": (
|
|
"bfbfef9099bf0e90480d48b3a1a741d583fc939284f869958e9c09e177098c73",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-compiler-for-linux/22-1/arm-compiler-for-linux_22.1_RHEL-7_aarch64.tar",
|
|
),
|
|
"RHEL-8": (
|
|
"28116f6030c95ee8f69eba89023966974d6b44d4a686098f5c3c03e34f7495f6",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-compiler-for-linux/22-1/arm-compiler-for-linux_22.1_RHEL-8_aarch64.tar",
|
|
),
|
|
"SLES-15": (
|
|
"6616dba1af4a73300ce822b645a0f1dfd363f507db5ea44cab1c6051ea388554",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-compiler-for-linux/22-1/arm-compiler-for-linux_22.1_SLES-15_aarch64.tar",
|
|
),
|
|
"Ubuntu-18.04": (
|
|
"3b3dd6f416299fbd14fbaf0b1bddf7e2f4445a186de7a87e9efdae0b9d0dc3d5",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-compiler-for-linux/22-1/arm-compiler-for-linux_22.1_Ubuntu-18.04_aarch64.tar",
|
|
),
|
|
"Ubuntu-20.04": (
|
|
"e6361a08f75817c8dbfb56dc72578810eaf5ffb65591215e394cb3ec6bdd9c10",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-compiler-for-linux/22-1/arm-compiler-for-linux_22.1_Ubuntu-20.04_aarch64.tar",
|
|
),
|
|
},
|
|
"22.0.2": {
|
|
"RHEL-7": (
|
|
"e4dec577ed2d33124a556ba05584fad45a9acf6e13dccadb37b521d1bad5a826",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-2/arm-compiler-for-linux_22.0.2_RHEL-7_aarch64.tar",
|
|
),
|
|
"RHEL-8": (
|
|
"3064bec6e0e3d4da9ea2bcdcb4590a8fc1f7e0e97092e24e2245c7f1745ef4f3",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-2/arm-compiler-for-linux_22.0.2_RHEL-8_aarch64.tar",
|
|
),
|
|
"SLES-15": (
|
|
"82dea469dc567b848bcaa6cbaed3eb3faaf45ceb9ec7071bdfef8a383e929ef8",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-2/arm-compiler-for-linux_22.0.2_SLES-15_aarch64.tar",
|
|
),
|
|
"Ubuntu-18.04": (
|
|
"355f548e86b9fa90d72684480d13ec60e6bec6b2bd837df42ac84d5a8fdebc48",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-2/arm-compiler-for-linux_22.0.2_Ubuntu-18.04_aarch64.tar",
|
|
),
|
|
"Ubuntu-20.04": (
|
|
"a2a752dce089a34b91dc89c0d1dd8b58a4104bf7c9ba3affd71fd1fd593e3732",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-2/arm-compiler-for-linux_22.0.2_Ubuntu-20.04_aarch64.tar",
|
|
),
|
|
},
|
|
"22.0.1": {
|
|
"RHEL-7": (
|
|
"6b0ab76dce3fd44aab1e679baef01367c86f6bbd3544e04f9642b6685482cd76",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-1/arm-compiler-for-linux_22.0.1_RHEL-7_aarch64.tar",
|
|
),
|
|
"RHEL-8": (
|
|
"41e5bffc52701b1e8a606f8db09c3c02e35ae39eae0ebeed5fbb41a13e61f057",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-1/arm-compiler-for-linux_22.0.1_RHEL-8_aarch64.tar",
|
|
),
|
|
"SLES-15": (
|
|
"b578ff517dec7fa23c4b7353a1a7c958f28cc9c9447f71f7c4e83de2e2c5538f",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-1/arm-compiler-for-linux_22.0.1_SLES-15_aarch64.tar",
|
|
),
|
|
"Ubuntu-18.04": (
|
|
"becc6826ce0f6e696092e79a40f758d7cd0302227f6cfc7c2215f6483ade9748",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-1/arm-compiler-for-linux_22.0.1_Ubuntu-18.04_aarch64.tar",
|
|
),
|
|
"Ubuntu-20.04": (
|
|
"dea136238fc2855c41b8a8154bf279b7df5df8dba48d8f29121fa26f343e7cdb",
|
|
"https://developer.arm.com/-/media/Files/downloads/hpc/arm-allinea-studio/22-0-1/arm-compiler-for-linux_22.0.1_Ubuntu-20.04_aarch64.tar",
|
|
),
|
|
},
|
|
}
|
|
|
|
|
|
def get_os():
|
|
spack_os = spack.platforms.host().default_os
|
|
return _os_map.get(spack_os, "RHEL-7")
|
|
|
|
|
|
def get_armpl_version_to_3(spec):
|
|
"""Return version string with 3 numbers"""
|
|
version_len = len(spec.version)
|
|
assert version_len == 2 or version_len == 3
|
|
if version_len == 2:
|
|
return spec.version.string + ".0"
|
|
elif version_len == 3:
|
|
return spec.version.string
|
|
|
|
|
|
def get_armpl_prefix(spec):
|
|
return join_path(
|
|
spec.prefix,
|
|
"armpl-{}_AArch64_{}_arm-linux-compiler_aarch64-linux".format(
|
|
get_armpl_version_to_3(spec), get_os()
|
|
),
|
|
)
|
|
|
|
|
|
def get_acfl_prefix(spec):
|
|
acfl_prefix = spec.prefix
|
|
return join_path(
|
|
acfl_prefix,
|
|
"arm-linux-compiler-{0}_Generic-AArch64_{1}_aarch64-linux".format(spec.version, get_os()),
|
|
)
|
|
|
|
|
|
class Acfl(Package):
|
|
"""Arm Compiler combines the optimized tools and libraries from Arm
|
|
with a modern LLVM-based compiler framework.
|
|
"""
|
|
|
|
homepage = "https://developer.arm.com/tools-and-software/server-and-hpc/arm-allinea-studio"
|
|
url = "https://developer.arm.com/-/media/Files/downloads/hpc/arm-compiler-for-linux/22-1/arm-compiler-for-linux_22.1_Ubuntu-20.04_aarch64.tar"
|
|
|
|
maintainers = ["annop-w"]
|
|
|
|
# Build Versions: establish OS for URL
|
|
acfl_os = get_os()
|
|
|
|
# Build Versions
|
|
for ver, packages in _versions.items():
|
|
pkg = packages.get(acfl_os)
|
|
if pkg:
|
|
version(ver, sha256=pkg[0], url=pkg[1])
|
|
|
|
# Only install for Aarch64
|
|
conflicts("target=x86_64:", msg="Only available on Aarch64")
|
|
conflicts("target=ppc64:", msg="Only available on Aarch64")
|
|
conflicts("target=ppc64le:", msg="Only available on Aarch64")
|
|
|
|
executables = [r"armclang", r"armclang\+\+", r"armflang"]
|
|
|
|
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")
|
|
|
|
# Licensing - Not required from 22.0.1 on.
|
|
|
|
# Run the installer with the desired install directory
|
|
def install(self, spec, prefix):
|
|
exe = Executable("./arm-compiler-for-linux_{0}_{1}.sh".format(spec.version, get_os()))
|
|
exe("--accept", "--force", "--install-to", prefix)
|
|
|
|
@classmethod
|
|
def determine_version(cls, exe):
|
|
regex_str = r"Arm C\/C\+\+\/Fortran Compiler version ([\d\.]+) " r"\(build number (\d+)\) "
|
|
version_regex = re.compile(regex_str)
|
|
try:
|
|
output = spack.compiler.get_compiler_version_output(exe, "--version")
|
|
match = version_regex.search(output)
|
|
if match:
|
|
if match.group(1).count(".") == 1:
|
|
return match.group(1) + ".0." + match.group(2)
|
|
return match.group(1) + "." + match.group(2)
|
|
except spack.util.executable.ProcessError:
|
|
pass
|
|
except Exception as e:
|
|
tty.debug(e)
|
|
|
|
@classmethod
|
|
def determine_variants(cls, exes, version_str):
|
|
compilers = {}
|
|
for exe in exes:
|
|
if "armclang" in exe:
|
|
compilers["c"] = exe
|
|
if "armclang++" in exe:
|
|
compilers["cxx"] = exe
|
|
if "armflang" in exe:
|
|
compilers["fortran"] = exe
|
|
return "", {"compilers": compilers}
|
|
|
|
@property
|
|
def cc(self):
|
|
msg = "cannot retrieve C compiler [spec is not concrete]"
|
|
assert self.spec.concrete, msg
|
|
if self.spec.external:
|
|
return self.spec.extra_attributes["compilers"].get("c", None)
|
|
return join_path(get_acfl_prefix(self.spec), "bin", "armclang")
|
|
|
|
@property
|
|
def cxx(self):
|
|
msg = "cannot retrieve C++ compiler [spec is not concrete]"
|
|
assert self.spec.concrete, msg
|
|
if self.spec.external:
|
|
return self.spec.extra_attributes["compilers"].get("cxx", None)
|
|
return join_path(get_acfl_prefix(self.spec), "bin", "armclang++")
|
|
|
|
@property
|
|
def fortran(self):
|
|
msg = "cannot retrieve Fortran compiler [spec is not concrete]"
|
|
assert self.spec.concrete, msg
|
|
if self.spec.external:
|
|
return self.spec.extra_attributes["compilers"].get("fortran", None)
|
|
return join_path(get_acfl_prefix(self.spec), "bin", "armflang")
|
|
|
|
@property
|
|
def lib_suffix(self):
|
|
suffix = ""
|
|
suffix += "_ilp64" if self.spec.satisfies("+ilp64") else ""
|
|
suffix += "_mp" if self.spec.satisfies("threads=openmp") else ""
|
|
return suffix
|
|
|
|
@property
|
|
def blas_libs(self):
|
|
armpl_prefix = get_armpl_prefix(self.spec)
|
|
|
|
libname = "libarmpl" + self.lib_suffix
|
|
|
|
# Get ArmPL Lib
|
|
armpl_libs = find_libraries(
|
|
[libname, "libamath", "libastring"],
|
|
root=armpl_prefix,
|
|
shared=self.spec.satisfies("+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" + self.lib_suffix
|
|
|
|
incdir = join_path(armpl_dir, suffix)
|
|
|
|
hlist = find_all_headers(incdir)
|
|
hlist.directories = [incdir]
|
|
return hlist
|
|
|
|
def setup_run_environment(self, env):
|
|
arm_dir = get_acfl_prefix(self.spec)
|
|
armpl_dir = get_armpl_prefix(self.spec)
|
|
env.set("ARM_LINUX_COMPILER_DIR", arm_dir)
|
|
env.set("ARM_LINUX_COMPILER_INCLUDES", join_path(arm_dir, "includes"))
|
|
env.prepend_path("LD_LIBRARY_PATH", join_path(arm_dir, "lib"))
|
|
env.prepend_path("LD_LIBRARY_PATH", join_path(armpl_dir, "lib"))
|
|
env.prepend_path("PATH", join_path(arm_dir, "bin"))
|
|
env.prepend_path("CPATH", join_path(arm_dir, "include"))
|
|
env.prepend_path("MANPATH", join_path(arm_dir, "share", "man"))
|
|
|
|
@run_after("install")
|
|
def check_install(self):
|
|
armpl_dir = get_armpl_prefix(self.spec)
|
|
armpl_example_dir = join_path(armpl_dir, "examples")
|
|
# run example makefile
|
|
make("-C", armpl_example_dir, "CC=" + self.cc, "F90=" + self.fortran)
|
|
# clean up
|
|
make("-C", armpl_example_dir, "clean")
|