armcomputelibrary: new package (#35515)
This commit is contained in:
Annop Wongwathanarat
GitHub
parent
65ee062bf6
commit
73fcda19d2
1 changed files with 192 additions and 0 deletions
192
var/spack/repos/builtin/packages/armcomputelibrary/package.py
Normal file
192
var/spack/repos/builtin/packages/armcomputelibrary/package.py
Normal file
|
|
@ -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),
|
||||
)
|
||||
Loading…
Reference in a new issue