Skip to content

Commit 8e6415c

Browse files
Add cray-mpich with working GTL for rocm>=7.1 (#1250)
* Update system.py * Update system.py * Update systems/llnl-elcapitan/system.py Co-authored-by: Nathan Hanford <nhanford@llnl.gov> * lint --------- Co-authored-by: Nathan Hanford <nhanford@llnl.gov>
1 parent 31b4e50 commit 8e6415c

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

systems/llnl-elcapitan/system.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -140,8 +140,8 @@ class LlnlElcapitan(System):
140140
"6.4.1",
141141
"6.4.2",
142142
"6.4.3",
143-
"7.0.1",
144143
"7.1.0",
144+
"7.2.0",
145145
),
146146
description="ROCm version",
147147
)
@@ -222,7 +222,11 @@ def __init__(self, spec):
222222
f"{self.gcc_version.major}.{self.gcc_version.minor}"
223223
)
224224
else:
225-
if self.rocm_version >= Version("6.4.0"):
225+
if self.rocm_version >= Version("7.1.0"):
226+
self.cce_version = Version("21.0.0")
227+
self.mpi_version = Version("9.1.0")
228+
# No working self.rccl_version
229+
elif self.rocm_version >= Version("6.4.0"):
226230
self.cce_version = Version("20.0.0")
227231
self.mpi_version = Version("9.0.1")
228232
self.rccl_version = Version("6.4.1")
@@ -234,7 +238,11 @@ def __init__(self, spec):
234238
self.cce_version = Version("16.0.0")
235239
self.mpi_version = Version("8.1.26")
236240
self.rccl_version = Version("5.4.3")
237-
if self.rocm_version >= Version("6.4.0"):
241+
if self.rocm_version >= Version("7.1.0"):
242+
self.pmi_version = Version("6.1.16")
243+
self.pals_version = Version("1.2.12")
244+
self.llvm_version = Version("20.0.0")
245+
elif self.rocm_version >= Version("6.4.0"):
238246
self.pmi_version = Version("6.1.15.6")
239247
self.pals_version = Version("1.2.12")
240248
self.llvm_version = Version("19.0.0")
@@ -399,9 +407,7 @@ def compute_packages_section(self):
399407
},
400408
"mpi": {"require": "cray-mpich-gtl"},
401409
"libfabric": {
402-
"externals": [
403-
{"spec": "libfabric@2.1", "prefix": "/opt/cray/libfabric/2.1"}
404-
],
410+
"externals": [{"spec": "libfabric@2.3.1", "prefix": "/usr"}],
405411
"buildable": False,
406412
},
407413
"ncurses": {

0 commit comments

Comments
 (0)