4d563acd1b
* bump up the version for ROCm-6.4.0 * update rocm-openmp-extras * fix for rocprofiler-sdk * fix hipblaslt ci fail * bump hipsparselt * set clang to use llvm-amdgpu * modify rom-smi-lib and update hip-tensor * miopen-hip: replace patch with commit * miopen-hip: remove patch * fix rocdecode libdrm error * rocjpeg fix libdrm error * rocm-bandwidth-test and amdsmi: add c dependency * add explicit roctracer-dev dependency * fix issue with rocm-openmp-extras using external * add c dependency to rocm-debug-agent and rocm-dbgapi * rocm-debug-agent: set clang as compiler * hip-tensor: add c dependency * hipsparselt: modify patch * hipsparselt: fix ROCM_SMI_PATH * hipsparselt: add c dependency * rocm-validation-suite: add patch for libdrm * rdc: remove rocm-smi-lib dependency * rocmwmma: replace patch * rocmwmma: remove old patch * rocwmma: modify patch * add c dependencies * add c dependency to rocwmma and rocm-examples * rocAL: force use of spack python * rdc: add c dependency * roctracer-dev: use llvm-amdgpu clang++ * rocm-tensile: add patch * rocm-cmake: fix standalone test * rocblas: fix client test error * roctracer-dev: fix ci fail * hipblaslt: add patch for client test * hipfort: add c dependency * py-tensorflow restrict to rocm 6.3
152 lines
6.4 KiB
Python
152 lines
6.4 KiB
Python
# Copyright Spack Project Developers. See COPYRIGHT file for details.
|
|
#
|
|
# SPDX-License-Identifier: (Apache-2.0 OR MIT)
|
|
|
|
from spack_repo.builtin.build_systems.cmake import CMakePackage
|
|
from spack_repo.builtin.build_systems.rocm import ROCmPackage
|
|
|
|
from spack.package import *
|
|
|
|
|
|
class Hipblaslt(CMakePackage):
|
|
"""hipBLASLt is a library that provides general matrix-matrix operations with a flexible API
|
|
and extends functionalities beyond a traditional BLAS library"""
|
|
|
|
homepage = "https://github.com/ROCm/hipBLASLt"
|
|
url = "https://github.com/ROCm/hipBLASLt/archive/refs/tags/rocm-6.1.2.tar.gz"
|
|
git = "https://github.com/ROCm/hipBLASLt.git"
|
|
|
|
maintainers("srekolam", "afzpatel", "renjithravindrankannath")
|
|
|
|
license("MIT")
|
|
version("6.4.0", sha256="a4baa0c7336db9d46a0884c8ccfd0fb7e00a502b478aed9f588aa26fa8773353")
|
|
version("6.3.3", sha256="f32d666b37bdbecbf924cc98653fa3d30a0de629039d4dad44d35a2082e39e5a")
|
|
version("6.3.2", sha256="cc4875b1a5cf1708a7576c42aff6b4cb790cb7337f5dc2df33119a4aadcef027")
|
|
version("6.3.1", sha256="9a18a2e44264a21cfe58ed102fd3e34b336f23d6c191ca2da726e8e0883ed663")
|
|
version("6.3.0", sha256="e570996037ea42eeca4c9b8b0b77a202d40be1a16068a6245595c551d80bdcad")
|
|
version("6.2.4", sha256="b8a72cb1ed4988b0569817c6387fb2faee4782795a0d8f49b827b32b52572cfd")
|
|
version("6.2.1", sha256="9b062b1d6d945349c31828030c8c1d99fe57d14a1837196ff9aa67bf10ef43f1")
|
|
version("6.2.0", sha256="aec9edc75ae4438aa712192c784e2bed683d2839b502b6aadb18f6012306749b")
|
|
version("6.1.2", sha256="fcfe950f7b87c421565abe090b2de6f463afc1549841002f105ecca7bbbf59e5")
|
|
version("6.1.1", sha256="1e21730ade59b5e32432fa0981383f689a380b1ffc92fe950822722da9521a72")
|
|
version("6.1.0", sha256="90fc2f2c9e11c87e0529e824e4b0561dbc850f8ffa21be6932ae63cbaa27cdf0")
|
|
version("6.0.2", sha256="e281a1a7760fab8c3e0baafe17950cf43c422184e3226e3c14eb06e50c69d421")
|
|
version("6.0.0", sha256="6451b6fdf7f24787628190bbe8f2208c929546b68b692d8355d2f18bea7ca7db")
|
|
|
|
amdgpu_targets = ROCmPackage.amdgpu_targets
|
|
|
|
variant(
|
|
"amdgpu_target",
|
|
description="AMD GPU architecture",
|
|
values=auto_or_any_combination_of(*amdgpu_targets),
|
|
sticky=True,
|
|
)
|
|
variant("asan", default=False, description="Build with address-sanitizer enabled or disabled")
|
|
|
|
depends_on("c", type="build")
|
|
depends_on("cxx", type="build")
|
|
depends_on("cmake@3.25.2:", type="build", when="@6.2.0:")
|
|
depends_on("python@3.7:")
|
|
depends_on("python@3.8:", when="@6.4:")
|
|
|
|
for ver in [
|
|
"6.0.0",
|
|
"6.0.2",
|
|
"6.1.0",
|
|
"6.1.1",
|
|
"6.1.2",
|
|
"6.2.0",
|
|
"6.2.1",
|
|
"6.2.4",
|
|
"6.3.0",
|
|
"6.3.1",
|
|
"6.3.2",
|
|
"6.3.3",
|
|
"6.4.0",
|
|
]:
|
|
depends_on(f"hip@{ver}", when=f"@{ver}")
|
|
depends_on(f"llvm-amdgpu@{ver}", when=f"@{ver}")
|
|
depends_on(f"rocm-openmp-extras@{ver}", type="test", when=f"@{ver}")
|
|
|
|
for ver in ["6.0.0", "6.0.2", "6.1.0", "6.1.1", "6.1.2", "6.2.0", "6.2.1", "6.2.4"]:
|
|
depends_on(f"hipblas@{ver}", when=f"@{ver}")
|
|
|
|
for ver in ["6.3.0", "6.3.1", "6.3.2", "6.3.3", "6.4.0"]:
|
|
depends_on(f"hipblas-common@{ver}", when=f"@{ver}")
|
|
depends_on(f"rocm-smi-lib@{ver}", when=f"@{ver}")
|
|
|
|
for ver in ["6.4.0"]:
|
|
depends_on(f"roctracer-dev@{ver}", when=f"@{ver}")
|
|
|
|
depends_on("msgpack-c")
|
|
depends_on("py-joblib", type=("build", "link"))
|
|
depends_on("googletest@1.10.0:", type="test")
|
|
depends_on("netlib-lapack@3.7.1:", type="test")
|
|
depends_on("py-pyyaml", type="test")
|
|
depends_on("python-venv", when="@6.4:")
|
|
|
|
# Sets the proper for clang++ and clang-offload-blunder.
|
|
# Also adds hipblas and msgpack include directories
|
|
patch("001_Set_LLVM_Paths_And_Add_Includes.patch", when="@6.0")
|
|
# Below patch sets the proper path for clang++ and clang-offload-blunder.
|
|
# Also adds hipblas and msgpack include directories for 6.1.0 release.
|
|
patch("0001-Set-LLVM_Path-Add-Hiblas-Include-to-CmakeLists-6.1.Patch", when="@6.1:6.2")
|
|
patch("0001-Set-LLVM-Path-6.3.Patch", when="@6.3")
|
|
patch("002-link-roctracer.patch", when="@6.4")
|
|
patch("003-use-rocm-smi-config.patch", when="@6.4")
|
|
|
|
def setup_build_environment(self, env: EnvironmentModifications) -> None:
|
|
env.set("CXX", self.spec["hip"].hipcc)
|
|
if self.spec.satisfies("@6.3.0:"):
|
|
env.set(
|
|
"TENSILE_ROCM_ASSEMBLER_PATH", f"{self.spec['llvm-amdgpu'].prefix}/bin/clang++"
|
|
)
|
|
env.set(
|
|
"TENSILE_ROCM_OFFLOAD_BUNDLER_PATH",
|
|
f"{self.spec['llvm-amdgpu'].prefix}/bin/clang-offload-bundler",
|
|
)
|
|
env.set("ROCM_SMI_PATH", f"{self.spec['rocm-smi-lib'].prefix}/bin/rocm-smi")
|
|
env.set(
|
|
"ROCM_AGENT_ENUMERATOR_PATH",
|
|
f"{self.spec['rocminfo'].prefix}/bin/rocm_agent_enumerator",
|
|
)
|
|
|
|
def patch(self):
|
|
if self.spec.satisfies("@6.3:"):
|
|
filter_file(
|
|
"${rocm_path}/llvm/bin",
|
|
self.spec["llvm-amdgpu"].prefix.bin,
|
|
"tensilelite/Tensile/Ops/gen_assembly.sh",
|
|
string=True,
|
|
)
|
|
filter_file(
|
|
"${rocm_path}/bin/amdclang++",
|
|
f'{self.spec["llvm-amdgpu"].prefix}/bin/amdclang++',
|
|
"library/src/amd_detail/rocblaslt/src/kernels/compile_code_object.sh",
|
|
string=True,
|
|
)
|
|
if self.spec.satisfies("@6.3:"):
|
|
filter_file(
|
|
"${rocm_path}/bin/amdclang++",
|
|
f'{self.spec["llvm-amdgpu"].prefix}/bin/amdclang++',
|
|
"tensilelite/Tensile/Ops/gen_assembly.sh",
|
|
string=True,
|
|
)
|
|
|
|
def cmake_args(self):
|
|
args = [
|
|
self.define("Tensile_CODE_OBJECT_VERSION", "default"),
|
|
self.define("MSGPACK_DIR", self.spec["msgpack-c"].prefix),
|
|
self.define_from_variant("ADDRESS_SANITIZER", "asan"),
|
|
self.define("BUILD_CLIENTS_TESTS", self.run_tests),
|
|
]
|
|
if "auto" not in self.spec.variants["amdgpu_target"]:
|
|
args.append(self.define_from_variant("AMDGPU_TARGETS", "amdgpu_target"))
|
|
if self.run_tests:
|
|
args.append(
|
|
self.define("ROCM_OPENMP_EXTRAS_DIR", self.spec["rocm-openmp-extras"].prefix)
|
|
)
|
|
if self.spec.satisfies("@6.4:"):
|
|
args.append(self.define("Python_ROOT", self.spec["python"].prefix.bin))
|
|
return args
|