73 lines
3.1 KiB
Python
73 lines
3.1 KiB
Python
# 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 NimrodAai(CMakePackage):
|
|
"""NIMROD infrastructure for FEM evaluation, integration and linear algebra
|
|
solves with device accelerated computing through OpenACC and abstract types
|
|
enabled by modern Fortran.
|
|
"""
|
|
|
|
homepage = "https://gitlab.com/NIMRODteam/nimrod-abstract"
|
|
url = (
|
|
"https://gitlab.com/NIMRODteam/nimrod-abstract/-/archive/23.9/nimrod-abstract-23.9.tar.gz"
|
|
)
|
|
git = "https://gitlab.com/NIMRODteam/nimrod-abstract.git"
|
|
|
|
maintainers("jacobrking")
|
|
|
|
version("main", branch="main")
|
|
version("23.9", sha256="212d591c5a5e7a394b56a5cf2f92cc69feafc49dd5f042fa95eeb6441649390b")
|
|
version("23.6", sha256="1794b89a5a64ff2b3c548818b90d17eef85d819ba4f63a76c41a682d5b76c14f")
|
|
|
|
variant("debug", default=False, description="Whether to enable debug code")
|
|
variant("openacc", default=False, description="Whether to enable OpenACC")
|
|
variant(
|
|
"openacc_autocompare", default=False, description="Whether to enable OpenACC autocompare"
|
|
)
|
|
variant("enable_shared", default=True, description="Whether to build and use shared libraries")
|
|
variant("mpi", default=False, description="Whether to enable MPI")
|
|
variant("time_level1", default=False, description="Whether to add timings at level 1")
|
|
variant("time_level2", default=False, description="Whether to add timings at level 2")
|
|
variant("nvtx_profile", default=False, description="Whether to enable NVTX profiling")
|
|
variant("openacc_cc", default="native", description="OpenACC compute capability")
|
|
variant(
|
|
"trap_fp_exceptions",
|
|
default=False,
|
|
description="Whether to enable trapping of floating point exceptions",
|
|
)
|
|
|
|
depends_on("cmake", type="build")
|
|
depends_on("mpi", when="+mpi")
|
|
depends_on("hdf5+fortran~mpi", type="build", when="~mpi")
|
|
depends_on("hdf5+fortran+mpi", type="build", when="+mpi")
|
|
|
|
def cmake_args(self):
|
|
args = [
|
|
self.define_from_variant("DEBUG", "debug"),
|
|
self.define_from_variant("ENABLE_OPENACC", "openacc"),
|
|
self.define_from_variant("ENABLE_MPI", "mpi"),
|
|
self.define_from_variant("ENABLE_SHARED", "enable_shared"),
|
|
self.define_from_variant("NVTX_PROFILE", "nvtx_profile"),
|
|
self.define_from_variant("TIME_LEVEL1", "time_level1"),
|
|
self.define_from_variant("TIME_LEVEL2", "time_level2"),
|
|
self.define_from_variant("TRAP_FP_EXCEPTIONS", "trap_fp_exceptions"),
|
|
]
|
|
if "+openacc" in self.spec:
|
|
addl_args = [
|
|
self.define_from_variant("ENABLE_OPENACC_AUTOCOMPARE", "openacc_autocompare"),
|
|
self.define_from_variant("OPENACC_CC", "openacc_cc"),
|
|
]
|
|
args.append(addl_args)
|
|
return args
|
|
|
|
@run_after("build")
|
|
@on_package_attributes(run_tests=True)
|
|
def check(self):
|
|
with working_dir(self.builder.build_directory):
|
|
ctest("--output-on-failure")
|