diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/armcomputelibrary/package.py | 192 |
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), + ) |