Skip to content

Commit 63d8490

Browse files
committed
Adapt sanity checks for AMD ROCm
1 parent 71b0102 commit 63d8490

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

easybuild/easyblocks/o/openmpi.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -218,11 +218,11 @@ def sanity_check_step(self):
218218
expected['mpif90'] = 'pgfortran'
219219
# for Clang the pattern is always clang
220220
for key in ['mpicc', 'mpicxx']:
221-
if expected[key] in ['clang++']:
221+
if expected[key] in ['clang++', 'amdclang', 'amdclang++']:
222222
expected[key] = 'clang'
223223
# for flang/flang-new the pattern is always flang
224224
for key in ['mpifort', 'mpif90']:
225-
if expected[key] in ['flang', 'flang-new']:
225+
if expected[key] in ['flang', 'flang-new', 'amdflang']:
226226
expected[key] = 'flang'
227227

228228
custom_commands = ["%s --version | grep '%s'" % (key, expected[key]) for key in sorted(expected.keys())]
@@ -265,4 +265,8 @@ def sanity_check_step(self):
265265
params['nr_ranks'] = 1
266266
custom_commands.append(mpi_cmd_tmpl % params)
267267

268+
rocmroot = get_software_root('ROCm')
269+
if rocmroot:
270+
custom_commands.append("ompi_info | grep -i 'rocm'")
271+
268272
super().sanity_check_step(custom_paths=custom_paths, custom_commands=custom_commands)

0 commit comments

Comments
 (0)