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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
|
# Copyright 2013-2024 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 Occa(Package):
"""OCCA is an open-source (MIT license) library used to program current
multi-core/many-core architectures. Devices (such as CPUs, GPUs,
Intel's Xeon Phi, FPGAs, etc) are abstracted using an offload-model
for application development and programming for the devices is done
through a C-based (OKL) or Fortran-based kernel language (OFL).
OCCA gives developers the ability to target devices at run-time by
using run-time compilation for device kernels.
"""
homepage = "https://libocca.org"
git = "https://github.com/libocca/occa.git"
maintainers("v-dobrev", "dmed256")
license("MIT")
version("develop")
version("1.2.0", tag="v1.2.0", commit="18379073b6497f677a20bfeced95b511f82c3355")
version("1.1.0", tag="v1.1.0", commit="c8a587666a23e045f25dc871c3257364a5f6a7d5")
version("1.0.9", tag="v1.0.9", commit="ebdb659c804f91f1e0f32fd700f9fe229458033c")
version("1.0.8", tag="v1.0.8", commit="55264f6b3d426f160dcf1f768c42d16d3ec14676")
version(
"1.0.0-alpha.5", tag="v1.0.0-alpha.5", commit="882ed5f92a40e60a80721727c350557be0ce6373"
)
version("0.2.0", tag="v0.2.0", commit="2eceaa5706ad6cf3a1b153c1f2a8a2fffa2d5945")
version("0.1.0", tag="v0.1.0", commit="381e886886dc87823769c5f20d0ecb29dd117afa")
variant("cuda", default=True, description="Activates support for CUDA")
variant("openmp", default=True, description="Activates support for OpenMP")
variant("opencl", default=True, description="Activates support for OpenCL")
depends_on("cuda", when="+cuda")
conflicts("%gcc@6:", when="^cuda@:8")
conflicts("%gcc@7:", when="^cuda@:9")
def install(self, spec, prefix):
# The build environment is set by the 'setup_build_environment' method.
# Copy the source to the installation directory and build OCCA there.
install_tree(".", prefix)
make("-C", prefix)
if self.run_tests:
make("-C", prefix, "test", parallel=False)
def _setup_runtime_flags(self, s_env):
spec = self.spec
s_env.set("OCCA_DIR", self.prefix)
s_env.set("OCCA_CXX", self.compiler.cxx)
cxxflags = spec.compiler_flags["cxxflags"]
if cxxflags:
# Run-time compiler flags:
s_env.set("OCCA_CXXFLAGS", " ".join(cxxflags))
if "+cuda" in spec:
cuda_dir = spec["cuda"].prefix
# Run-time CUDA compiler:
s_env.set("OCCA_CUDA_COMPILER", join_path(cuda_dir, "bin", "nvcc"))
def setup_build_environment(self, env):
spec = self.spec
# The environment variable CXX is automatically set to the Spack
# compiler wrapper.
# The cxxflags, if specified, will be set by the Spack compiler wrapper
# while the environment variable CXXFLAGS will remain undefined.
# We define CXXFLAGS in the environment to tell OCCA to use the user
# specified flags instead of its defaults. This way the compiler will
# get the cxxflags twice - once from the Spack compiler wrapper and
# second time from OCCA - however, only the second one will be seen in
# the verbose output, so we keep both.
cxxflags = spec.compiler_flags["cxxflags"]
if cxxflags:
env.set("CXXFLAGS", " ".join(cxxflags))
# For the cuda, openmp, and opencl variants, set the environment
# variable OCCA_{CUDA,OPENMP,OPENCL}_ENABLED only if the variant is
# disabled. Otherwise, let OCCA autodetect what is available.
if "+cuda" in spec:
cuda_dir = spec["cuda"].prefix
cuda_libs_list = ["libcuda", "libcudart", "libOpenCL"]
cuda_libs = find_libraries(cuda_libs_list, cuda_dir, shared=True, recursive=True)
env.set("OCCA_INCLUDE_PATH", cuda_dir.include)
env.set("OCCA_LIBRARY_PATH", ":".join(cuda_libs.directories))
else:
env.set("OCCA_CUDA_ENABLED", "0")
# Disable hip autodetection for now since it fails on some machines.
env.set("OCCA_HIP_ENABLED", "0")
if "~opencl" in spec:
env.set("OCCA_OPENCL_ENABLED", "0")
if "~openmp" in spec:
env.set("OCCA_OPENMP_ENABLED", "0")
# Setup run-time environment for testing.
env.set("OCCA_VERBOSE", "1")
self._setup_runtime_flags(env)
def setup_run_environment(self, env):
# The 'env' is included in the Spack generated module files.
self._setup_runtime_flags(env)
def setup_dependent_build_environment(self, env, dependent_spec):
# Export OCCA_* variables for everyone using this package from within
# Spack.
self._setup_runtime_flags(env)
|