summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--var/spack/repos/builtin/packages/armcomputelibrary/package.py192
1 files changed, 192 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/armcomputelibrary/package.py b/var/spack/repos/builtin/packages/armcomputelibrary/package.py
new file mode 100644
index 0000000000..57a909a5f6
--- /dev/null
+++ b/var/spack/repos/builtin/packages/armcomputelibrary/package.py
@@ -0,0 +1,192 @@
+# 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)
+
+from spack.package import *
+
+target_arch_list = (
+ "armv7a",
+ "armv7a-hf",
+ "arm64-v8a",
+ "arm64-v8.2-a",
+ "x86_32",
+ "x86_64",
+ "armv8a",
+ "armv8.2-a",
+ "armv8.6-a",
+ "armv8r64",
+ "x86",
+)
+
+
+class Armcomputelibrary(SConsPackage):
+ """The Arm Compute Library is a collection of low-level machine learning functions optimized
+ for Arm® Cortex®-A, Arm® Neoverse® and Arm® Mali™ GPUs architectures.
+ The library provides superior performance to other open source alternatives and
+ immediate support for new Arm® technologies e.g. SVE2."""
+
+ homepage = "https://arm-software.github.io/ComputeLibrary/latest/"
+ url = "https://github.com/ARM-software/ComputeLibrary/archive/refs/tags/v22.11.zip"
+ git = "https://github.com/ARM-software/ComputeLibrary.git"
+
+ maintainers = ["annop-w"]
+
+ version("22.11", sha256="2f70f54d84390625222503ea38650c00c49d4b70bc86a6b9aeeebee9d243865f")
+ version("22.08", sha256="5d76d07406b105f0bdf74ef80263236cb03baf0ade882f2bf8446bbc239e0079")
+ version("22.05", sha256="8ff308448874c6b72c1ce8d9f28af41d8b47c8e5c43b8ccc069da744e3c0a421")
+ version("22.02", sha256="0c1fe30b24e78bf5ca313ee8a33ad95e0d2aaddf64d4518ecec6a95e4bfba6e5")
+
+ depends_on("scons@2.3:")
+
+ phases = ["build"]
+
+ variant(
+ "build_type",
+ default="release",
+ values=("release", "debug"),
+ )
+ variant(
+ "threads",
+ default="cppthreads",
+ values=("cppthreads", "openmp"),
+ description="Enable C++11 threads/OpenMP backend. OpenMP backend only"
+ "works when building with g++ and not clang++.",
+ )
+ variant(
+ "multi_isa",
+ default=False,
+ description="Build Multi ISA binary version of library." " Note works only for armv8.2-a.",
+ )
+ variant(
+ "target_arch",
+ default="armv7a",
+ values=target_arch_list,
+ description="Target Architecture. The x86_32 and x86_64 targets can only be"
+ " used with neon=0 and opencl=1.",
+ )
+ variant(
+ "sve",
+ default=False,
+ description="Build for SVE.",
+ )
+ variant(
+ "sve2",
+ default=False,
+ description="Build for SVE2.",
+ )
+ variant(
+ "neon",
+ default=True,
+ description="Enable Arm® Neon™ support",
+ )
+ variant(
+ "experimental_dynamic_fusion",
+ default=False,
+ description="Build the experimental dynamic fusion files.",
+ )
+ variant(
+ "experimental_fixed_format_kernels",
+ default=False,
+ description="Enable fixed format kernels for GEMM.",
+ )
+ variant(
+ "benchmark_examples",
+ default=False,
+ description="Build benchmark examples programs.",
+ )
+ variant(
+ "validate_examples",
+ default=False,
+ description="Build validate examples programs.",
+ )
+ variant(
+ "validation_tests",
+ default=False,
+ description="Build validation test programs.",
+ )
+ variant(
+ "benchmark_tests",
+ default=False,
+ description="Build benchmark test programs.",
+ )
+
+ def build_args(self, spec, prefix):
+ args = ["-j{0}".format(make_jobs)]
+
+ if spec.satisfies("build_type=debug"):
+ args.append("debug=1")
+ elif spec.satisfies("build_type=release"):
+ args.append("debug=0")
+
+ if spec.satisfies("threads=openmp"):
+ args.append("cppthreads=0")
+ args.append("openmp=1")
+ elif spec.satisfies("threads=cppthreads"):
+ args.append("cppthreads=1")
+ args.append("openmp=0")
+
+ arch_value = spec.variants["target_arch"].value
+ arch_value += "-sve" if spec.variants["sve"].value else ""
+ arch_value += "-sve2" if spec.variants["sve2"].value else ""
+ args.append("arch={}".format(arch_value))
+ args.append("multi_isa={}".format(int(spec.variants["multi_isa"].value)))
+ args.append("neon={}".format(int(spec.variants["neon"].value)))
+ args.append(
+ "experimental_dynamic_fusion={}".format(
+ int(spec.variants["experimental_dynamic_fusion"].value)
+ )
+ )
+ args.append(
+ "experimental_fixed_format_kernels={}".format(
+ int(spec.variants["experimental_fixed_format_kernels"].value)
+ )
+ )
+ args.append("benchmark_examples={}".format(int(spec.variants["benchmark_examples"].value)))
+ args.append("validate_examples={}".format(int(spec.variants["validate_examples"].value)))
+ args.append("validation_tests={}".format(int(spec.variants["validation_tests"].value)))
+ args.append("benchmark_tests={}".format(int(spec.variants["benchmark_tests"].value)))
+ args.append("build=native")
+ args.append("install_dir={0}".format(prefix))
+
+ return args
+
+ def setup_build_environment(self, env):
+ # Spack compiler wrapper inject -mcpu flag for some targets.
+ # This can conflict with -march set in scons script, so override it here.
+ env.set("SPACK_TARGET_ARGS", "")
+
+ @property
+ def libs(self):
+ acl_libs = find_libraries(
+ ["libarm_compute", "libarm_compute_core", "libarm_compute_graph"],
+ root=self.spec.prefix,
+ shared=True,
+ recursive=True,
+ )
+ return acl_libs
+
+ @property
+ def headers(self):
+ incdir = join_path(self.spec.prefix, "include")
+ hlist = find_all_headers(incdir)
+ hlist.directories = [incdir]
+ return hlist
+
+ @run_after("build")
+ def make_sym_link(self):
+ # oneDNN expects header and lib files for ACL in arm_compute/ and build/ directory
+ prefix = self.spec.prefix
+ symlink(join_path(prefix, "include/arm_compute"), join_path(prefix, "arm_compute"))
+ symlink(join_path(prefix, "lib"), join_path(prefix, "build"))
+
+ @run_after("build")
+ def copy_additional_header(self):
+ # copy some additional header files needed by oneDNN
+ cpuinfo_dir = "src/common/cpuinfo"
+ mkdirp(join_path(self.spec.prefix, cpuinfo_dir))
+ for f in ["CpuInfo.h", "CpuIsaInfo.h", "CpuModel.h"]:
+ copy(
+ join_path(self.stage.source_path, cpuinfo_dir, f),
+ join_path(self.spec.prefix, cpuinfo_dir, f),
+ )