35 lines
1.4 KiB
Python
35 lines
1.4 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 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-6.0.2.tar.gz"
|
||
tags = ["rocm"]
|
||
|
||
maintainers("srekolam", "afzpatel")
|
||
|
||
version("master", branch="master")
|
||
version("6.0.2", sha256="6e6e7530eabbd1fb28b83efa5a49c19a6642d40e1554224ebb1e0a5999045e27")
|
||
version("6.0.0", sha256="268d7f114784b7e824f89c21c65c2efedbb5486f09a356a56dca1b89bde1ef7a")
|
||
version("5.7.1", sha256="96743d4e695fe865aef4097ae31d9b4e42a2d5a92135a005b0d187d9c0b17645")
|
||
version("5.7.0", sha256="4b17f6d43b17fe2dc1d0c61e9663d4752006f7898cc94231206444a1663eb252")
|
||
|
||
variant("asan", default=False, description="Build with address-sanitizer enabled or disabled")
|
||
|
||
for ver in ["5.7.0", "5.7.1", "6.0.0", "6.0.2", "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)
|
||
if self.spec.satisfies("+asan"):
|
||
self.asan_on(env)
|