Update software root check to ROCm-LLVM

Author: zerefwayne

easybuild/easyblocks/o/openmpi.py

@@ -227,6 +227,10 @@ def sanity_check_step(self):
 
         custom_commands = ["%s --version | grep '%s'" % (key, expected[key]) for key in sorted(expected.keys())]
 
+        rocmroot = get_software_root('ROCm-LLVM')
+        if rocmroot:
+            custom_commands.append("ompi_info | grep -i 'rocm'")
+
         # Add minimal test program to sanity checks
         # Run with correct MPI launcher
         mpi_cmd_tmpl, params = get_mpi_cmd_template(toolchain.OPENMPI, {}, mpi_version=self.version)
@@ -265,8 +269,4 @@ def sanity_check_step(self):
                     params['nr_ranks'] = 1
                     custom_commands.append(mpi_cmd_tmpl % params)
 
-                rocmroot = get_software_root('ROCm')
-                if rocmroot:
-                    custom_commands.append("ompi_info | grep -i 'rocm'")
-
         super().sanity_check_step(custom_paths=custom_paths, custom_commands=custom_commands)