diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/hip-tensor/package.py | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/hip-tensor/package.py b/var/spack/repos/builtin/packages/hip-tensor/package.py new file mode 100644 index 0000000000..e925031945 --- /dev/null +++ b/var/spack/repos/builtin/packages/hip-tensor/package.py @@ -0,0 +1,28 @@ +# 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 HipTensor(CMakePackage, ROCmPackage): + """AMD’s C++ library for accelerating tensor primitives""" + + homepage = "https://github.com/ROCm/hipTensor" + git = "https://github.com/ROCm/hipTensor.git" + url = "https://github.com/ROCm/hipTensor/archive/refs/tags/rocm-5.7.0.tar.gz" + tags = ["rocm"] + + maintainers("srekolam", "afzpatel") + + version("master", branch="master") + version("5.7.1", sha256="96743d4e695fe865aef4097ae31d9b4e42a2d5a92135a005b0d187d9c0b17645") + version("5.7.0", sha256="4b17f6d43b17fe2dc1d0c61e9663d4752006f7898cc94231206444a1663eb252") + + for ver in ["5.7.0", "5.7.1", "master"]: + depends_on(f"composable-kernel@{ver}", when=f"@{ver}") + depends_on(f"rocm-cmake@{ver}", when=f"@{ver}") + + def setup_build_environment(self, env): + env.set("CXX", self.spec["hip"].hipcc) |