From 5cfacb8c55e9785cfd8b2063309f40a39cefbcc7 Mon Sep 17 00:00:00 2001
From: Tobias Melson <tobias.melson@mpcdf.mpg.de>
Date: Thu, 28 Nov 2024 09:51:30 +0100
Subject: [PATCH] Added dependency on system:rocm
---
mpcdf_common.py | 11 ++++++++++-
1 file changed, 10 insertions(+), 1 deletion(-)
diff --git a/mpcdf_common.py b/mpcdf_common.py
index 21430a2..e06248c 100644
--- a/mpcdf_common.py
+++ b/mpcdf_common.py
@@ -809,6 +809,13 @@ def openmpi_flavor_dependencies(flavor, distribution):
return ()
+def rocm_dependencies(rocm):
+ if rocm == "rocm_6_1":
+ return ()
+ r = rocm.replace("_", "-", 1).replace("_", ".", 1)
+ return (("system:rocm", r),)
+
+
def openmpi_flavor_kind(flavor):
import re
return re.sub("_[0-9_-]+$", "", flavor)
@@ -1085,7 +1092,9 @@ Macros:
for rocm in rocms:
for compiler in filter(valid_rocm, compilers + amds):
- repo(rocm + "_" + compiler, ((project, compiler),), rocm=True, rocm_repository=rocm,
+ repo(rocm + "_" + compiler,
+ rocm_dependencies(rocm) + ((project, compiler),),
+ rocm=True, rocm_repository=rocm,
additional_tags=("Prefer: mpcdf_" + rocm,))
for mpi in filter(partial(valid_mpi, compiler), filter(is_openmpi, mpis)):
repo(rocm + "_" + mpi + "_" + compiler,
--
GitLab