intel-oneapi-mkl: add hpcx-mpi to the list of supported MPI libs (#39625)

Co-authored-by: Loris Ercole <a-lercole@microsoft.com>
This commit is contained in:
Loris Ercole 2023-08-25 14:53:22 +02:00 committed by GitHub
parent 8a08f09ac0
commit 4502351659
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -201,14 +201,16 @@ def _find_mkl_libs(self, shared):
] ]
): ):
libs.append(self._xlp64_lib("libmkl_blacs_intelmpi")) libs.append(self._xlp64_lib("libmkl_blacs_intelmpi"))
elif any(self.spec.satisfies(m) for m in ["^openmpi", "mpi_family=openmpi"]): elif any(
self.spec.satisfies(m) for m in ["^openmpi", "^hpcx-mpi", "mpi_family=openmpi"]
):
libs.append(self._xlp64_lib("libmkl_blacs_openmpi")) libs.append(self._xlp64_lib("libmkl_blacs_openmpi"))
else: else:
raise RuntimeError( raise RuntimeError(
( (
"intel-oneapi-mpi +cluster requires one of ^intel-oneapi-mpi, " "intel-oneapi-mkl +cluster requires one of ^intel-oneapi-mpi, "
"^intel-mpi, ^mpich, ^cray-mpich, mpi_family=mpich, ^openmpi, " "^intel-mpi, ^mpich, ^cray-mpich, mpi_family=mpich, ^openmpi, "
"or mpi_family=openmpi" "^hpcx-mpi, or mpi_family=openmpi"
) )
) )