1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
|
# 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 *
class Arrayfire(CMakePackage, CudaPackage):
"""ArrayFire is a high performance software library for parallel computing
with an easy-to-use API. Its array based function set makes parallel
programming more accessible."""
homepage = "https://arrayfire.org/docs/index.htm"
git = "https://github.com/arrayfire/arrayfire.git"
maintainers("umar456")
license("FreeImage")
version("master")
version("3.8.1", commit="823e8e399fe8c120c6ec7ec75f09e6106b3074ca", tag="v3.8.1")
version(
"3.7.3", commit="59ac7b980d1ae124aae914fb29cbf086c948954d", submodules=True, tag="v3.7.3"
)
version(
"3.7.2", commit="218dd2c99300e77496239ade76e94b0def65d032", submodules=True, tag="v3.7.2"
)
version(
"3.7.0", commit="fbea2aeb6f7f2d277dcb0ab425a77bb18ed22291", submodules=True, tag="v3.7.0"
)
variant("forge", default=False, description="Enable graphics library")
variant("opencl", default=False, description="Enable OpenCL backend")
depends_on("boost@1.70:")
depends_on("fftw-api@3:")
depends_on("blas")
depends_on("cuda@7.5:", when="+cuda")
depends_on("cudnn", when="+cuda")
depends_on("opencl", when="+opencl")
depends_on("pocl+icd", when="^[virtuals=opencl] pocl")
# TODO add more opencl backends:
# currently only Cuda backend is enabled
# https://github.com/arrayfire/arrayfire/wiki/Build-Instructions-for-Linux#opencl-backend-dependencies
depends_on("fontconfig", when="+forge")
depends_on("glfw@3.1.4:", when="+forge")
conflicts("cuda_arch=none", when="+cuda", msg="CUDA architecture is required")
@property
def libs(self):
query_parameters = self.spec.last_query.extra_parameters
libraries = []
if "cpu" in query_parameters:
libraries.append("libafcpu")
if "cuda" in query_parameters and "+cuda" in self.spec:
libraries.append("libafcuda")
if "opencl" in query_parameters and "+opencl" in self.spec:
libraries.append("libafopencl")
if not query_parameters or "unified" in query_parameters:
libraries.append("libaf")
return find_libraries(libraries, root=self.prefix, recursive=True)
def cmake_args(self):
args = []
args.extend(
[
self.define_from_variant("AF_BUILD_CUDA", "cuda"),
self.define_from_variant("AF_BUILD_FORGE", "forge"),
self.define_from_variant("AF_BUILD_OPENCL", "opencl"),
self.define("BUILD_TESTING", self.run_tests),
]
)
if "+cuda" in self.spec:
arch_list = [
"{}.{}".format(arch[:-1], arch[-1])
for arch in self.spec.variants["cuda_arch"].value
]
args.append(self.define("CUDA_architecture_build_targets", arch_list))
if self.spec["blas"].name in INTEL_MATH_LIBRARIES:
if self.version >= Version("3.8.0"):
args.append(self.define("AF_COMPUTE_LIBRARY", "Intel-MKL"))
else:
args.append(self.define("USE_CPU_MKL", True))
args.append(self.define("USE_OPENCL_MKL", True))
if "%intel" not in self.spec:
args.append(self.define("MKL_THREAD_LAYER", "GNU OpenMP"))
else:
if self.version >= Version("3.8.0"):
args.append(self.define("AF_COMPUTE_LIBRARY", "FFTW/LAPACK/BLAS"))
else:
args.append(self.define("USE_CPU_MKL", False))
args.append(self.define("USE_OPENCL_MKL", False))
return args
|