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

add ck variant to miopen-hip (#49143)

Browse Source
This commit is contained in:
afzpatel 2025-02-25 07:48:49 -05:00 committed by GitHub
parent 9e508b0321
commit 93329d7f99
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
var/spack/repos/builtin/packages/miopen-hip/package.py
View File

@ -43,6 +43,11 @@ class MiopenHip(CMakePackage):
version("5.3.3", sha256="7efc98215d23a2caaf212378c37e9a6484f54a4ed3e9660719286e4f287d3715")
version("5.3.0", sha256="c5819f593d71beeda2eb24b89182912240cc40f83b2b8f9de695a8e230aa4ea6")
variant(
"ck",
default=True,
description="Enable MIOpen to use composable kernels for various operation",
)
variant("asan", default=False, description="Build with address-sanitizer enabled or disabled")
conflicts("+asan", when="os=rhel9")
@ -143,10 +148,8 @@ class MiopenHip(CMakePackage):
"6.3.1",
"6.3.2",
]:
depends_on("nlohmann-json", type="link")
depends_on(f"composable-kernel@{ver}", when=f"@{ver}")
depends_on(f"composable-kernel@{ver}", when=f"@{ver} +ck")
for ver in ["5.4.0", "5.4.3", "5.5.0"]:
depends_on("nlohmann-json", type="link")
depends_on(f"rocmlir@{ver}", when=f"@{ver}")
for ver in [
"6.0.0",
@ -162,14 +165,15 @@ class MiopenHip(CMakePackage):
"6.3.2",
]:
depends_on(f"roctracer-dev@{ver}", when=f"@{ver}")
for ver in ["6.1.0", "6.1.1", "6.1.2"]:
depends_on("googletest")
for ver in ["6.2.0", "6.2.1", "6.2.4", "6.3.0", "6.3.1", "6.3.2"]:
depends_on(f"rocrand@{ver}", when=f"@{ver}")
for ver in ["6.3.0", "6.3.1", "6.3.2"]:
depends_on(f"hipblas@{ver}", when=f"@{ver}")
depends_on(f"hipblaslt@{ver}", when=f"@{ver}")
depends_on("nlohmann-json", type="link")
depends_on("googletest", when="@6.1:")
def setup_build_environment(self, env):
lib_dir = self.spec["zlib-api"].libs.directories[0]
env.prepend_path("LIBRARY_PATH", lib_dir)
@ -224,7 +228,7 @@ def cmake_args(self):
args.append(self.define("MIOPEN_USE_MLIR", "ON"))
args.append(self.define("MIOPEN_ENABLE_AI_KERNEL_TUNING", "OFF"))
if self.spec.satisfies("@5.5.1:"):
args.append(self.define("MIOPEN_USE_COMPOSABLEKERNEL", "ON"))
args.append(self.define_from_variant("MIOPEN_USE_COMPOSABLEKERNEL", "ck"))
args.append(self.define("MIOPEN_ENABLE_AI_KERNEL_TUNING", "OFF"))
args.append(self.define("MIOPEN_USE_MLIR", "OFF"))
if self.spec.satisfies("@5.7.0:"):