summaryrefslogtreecommitdiff
path: root/var/spack/repos/builtin/packages/hiptt/package.py
blob: 2e3c4704c273848eb515000aed3c5e40f7e3e5ea (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
# 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 Hiptt(MakefilePackage, ROCmPackage):
    """hipTT - Fast GPU Tensor Transpose for NVIDIA and AMD GPU."""

    homepage = "https://github.com/DmitryLyakh/hipTT"
    url = "https://github.com/DmitryLyakh/hipTT"
    git = "https://github.com/DmitryLyakh/hipTT.git"
    tags = ["ecp", "radiuss"]

    maintainers("bvanessen")

    license("Unlicense")

    version("master", branch="master")

    patch("bugfix_make.patch")

    # To enable this package add it to the LD_LIBRARY_PATH
    def setup_dependent_build_environment(self, env, dependent_spec):
        hiptt_home = self.spec["hiptt"].prefix
        env.prepend_path("cuTT_ROOT", hiptt_home)
        env.prepend_path("cuTT_LIBRARY", hiptt_home.lib)
        env.prepend_path("cuTT_INCLUDE_PATH", hiptt_home.include)

    def install(self, spec, prefix):
        mkdir(prefix.bin)
        install("bin/cutt_test", prefix.bin)
        install("bin/cutt_bench", prefix.bin)
        install_tree("lib", prefix.lib)
        install_tree("include", prefix.include)