zhangyiss/spack
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

hip-tensor: add new package (#40991)

Browse Source
This commit is contained in:
afzpatel 2024-01-09 12:39:10 -05:00 committed by GitHub
parent 705e0c6958
commit 097c04195a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
var/spack/repos/builtin/packages/hip-tensor/package.py Normal file
View File

@ -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)