summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/e3sm-kernels/package.py
blob: 0f1df42267b86b37f9e9fe03f78b11aeeda768ff (plain) (blame)
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
# 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)

import os

from spack.package import *


class E3smKernels(MakefilePackage):
    """
    Climate kernels for Co-design that originate from the Energy
    Exascale Earth System Model (E3SM).
    """

    homepage = "https://github.com/e3SM-Project/codesign-kernels"
    url = "https://github.com/E3SM-Project/codesign-kernels/archive/refs/tags/v1.0.tar.gz"
    git = "https://github.com/E3SM-Project/codesign-kernels.git"

    maintainers("sarats", "philipwjones")

    version("master", branch="master")
    version("1.0", sha256="358249785ba9f95616feecbb6f37f7694646568499c11b2094c9233999c6cc95")

    variant(
        "kernel",
        default="atmosphere",
        values=("atmosphere", "mmf-mpdata-tracer"),
        description="Specify E3SM Kernel to Build",
        multi=False,
    )

    @property
    def build_directory(self):
        return self.spec.variants["kernel"].value

    @property
    def build_targets(self):
        # Spack will provide optimization flags
        # But we still need to pass in fortran flags for gfortran
        args = []
        # Test for gfortran specifically due to hybrid compilers like llvm
        if "gfortran" in self.compiler.fc:
            args.append("FFLAGS=-ffree-line-length-none")
        return args

    def install(self, spec, prefix):
        # Manually copy binaries over
        mkdir(prefix.bin)
        if self.spec.variants["kernel"].value == "atmosphere":
            install(os.path.join("atmosphere", "atm"), prefix.bin.atm)
        elif self.spec.variants["kernel"].value == "mmf-mpdata-tracer":
            install(os.path.join("mmf-mpdata-tracer", "advect"), prefix.bin.advect)