# Copyright 2013-2024 Lawrence Livermore National Security, LLC and other # Spack Project Developers. See the top-level COPYRIGHT file for details. # # SPDX-License-Identifier: (Apache-2.0 OR MIT) from spack.package import * @IntelOneApiPackage.update_description class IntelOneapiMpi(IntelOneApiLibraryPackage): """Intel MPI Library is a multifabric message-passing library that implements the open-source MPICH specification. Use the library to create, maintain, and test advanced, complex applications that perform better on high-performance computing (HPC) clusters based on Intel processors. """ maintainers("rscohn2") homepage = "https://software.intel.com/content/www/us/en/develop/tools/oneapi/components/mpi-library.html" version( "2021.11.0", url="https://registrationcenter-download.intel.com/akdlm//IRC_NAS/2c45ede0-623c-4c8e-9e09-bed27d70fa33/l_mpi_oneapi_p_2021.11.0.49513_offline.sh", sha256="9a96caeb7abcf5aa08426216db38a2c7936462008b9825036266bc79cb0e30d8", expand=False, ) version( "2021.10.0", url="https://registrationcenter-download.intel.com/akdlm/IRC_NAS/4f5871da-0533-4f62-b563-905edfb2e9b7/l_mpi_oneapi_p_2021.10.0.49374_offline.sh", sha256="ab2e97d87b139201a2e7dab9a61ac6e8927b7783b459358c4ad69a1b1c064f40", expand=False, ) version( "2021.9.0", url="https://registrationcenter-download.intel.com/akdlm/IRC_NAS/718d6f8f-2546-4b36-b97b-bc58d5482ebf/l_mpi_oneapi_p_2021.9.0.43482_offline.sh", sha256="5c170cdf26901311408809ced28498b630a494428703685203ceef6e62735ef8", expand=False, ) version( "2021.8.0", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/19131/l_mpi_oneapi_p_2021.8.0.25329_offline.sh", sha256="0fcb1171fc42fd4b2d863ae474c0b0f656b0fa1fdc1df435aa851ccd6d1eaaf7", expand=False, ) version( "2021.7.1", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/19010/l_mpi_oneapi_p_2021.7.1.16815_offline.sh", sha256="90e7804f2367d457cd4cbf7aa29f1c5676287aa9b34f93e7c9a19e4b8583fff7", expand=False, ) version( "2021.7.0", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18926/l_mpi_oneapi_p_2021.7.0.8711_offline.sh", sha256="4eb1e1487b67b98857bc9b7b37bcac4998e0aa6d1b892b2c87b003bf84fb38e9", expand=False, ) version( "2021.6.0", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18714/l_mpi_oneapi_p_2021.6.0.602_offline.sh", sha256="e85db63788c434d43c1378e5e2bf7927a75d11aee8e6b78ee0d933da920977a6", expand=False, ) version( "2021.5.1", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18471/l_mpi_oneapi_p_2021.5.1.515_offline.sh", sha256="b992573959e39752e503e691564a0d876b099547c38b322d5775c5b06ec07a7f", expand=False, ) version( "2021.5.0", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18370/l_mpi_oneapi_p_2021.5.0.495_offline.sh", sha256="3aae53fe77f7c6aac7a32b299c25d6ca9a00ba4e2d512a26edd90811e59e7471", expand=False, ) version( "2021.4.0", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/18186/l_mpi_oneapi_p_2021.4.0.441_offline.sh", sha256="cc4b7072c61d0bd02b1c431b22d2ea3b84b967b59d2e587e77a9e7b2c24f2a29", expand=False, ) version( "2021.3.0", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/17947/l_mpi_oneapi_p_2021.3.0.294_offline.sh", sha256="04c48f864ee4c723b1b4ca62f2bea8c04d5d7e3de19171fd62b17868bc79bc36", expand=False, ) version( "2021.2.0", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/17729/l_mpi_oneapi_p_2021.2.0.215_offline.sh", sha256="d0d4cdd11edaff2e7285e38f537defccff38e37a3067c02f4af43a3629ad4aa3", expand=False, ) version( "2021.1.1", url="https://registrationcenter-download.intel.com/akdlm/irc_nas/17397/l_mpi_oneapi_p_2021.1.1.76_offline.sh", sha256="8b7693a156c6fc6269637bef586a8fd3ea6610cac2aae4e7f48c1fbb601625fe", expand=False, ) variant("ilp64", default=False, description="Build with ILP64 support") variant( "generic-names", default=False, description="Use generic names, e.g mpicc instead of mpiicc", ) variant( "external-libfabric", default=False, description="Enable external libfabric dependency" ) depends_on("libfabric", when="+external-libfabric", type=("link", "run")) provides("mpi@:3.1") @property def v2_layout_versions(self): return "@2021.11:" @property def component_dir(self): return "mpi" @property def env_script_args(self): if "+external-libfabric" in self.spec: return ("-i_mpi_ofi_internal=0",) else: return () def setup_dependent_package(self, module, dep_spec): if "+generic-names" in self.spec: self.spec.mpicc = join_path(self.component_prefix.bin, "mpicc") self.spec.mpicxx = join_path(self.component_prefix.bin, "mpicxx") self.spec.mpif77 = join_path(self.component_prefix.bin, "mpif77") self.spec.mpifc = join_path(self.component_prefix.bin, "mpifc") else: self.spec.mpicc = join_path(self.component_prefix.bin, "mpiicc") self.spec.mpicxx = join_path(self.component_prefix.bin, "mpiicpc") self.spec.mpif77 = join_path(self.component_prefix.bin, "mpiifort") self.spec.mpifc = join_path(self.component_prefix.bin, "mpiifort") def setup_dependent_build_environment(self, env, dependent_spec): env.set("I_MPI_CC", spack_cc) env.set("I_MPI_CXX", spack_cxx) env.set("I_MPI_F77", spack_f77) env.set("I_MPI_F90", spack_fc) env.set("I_MPI_FC", spack_fc) # Set compiler wrappers for dependent build stage if "+generic-names" in self.spec: env.set("MPICC", join_path(self.component_prefix.bin, "mpicc")) env.set("MPICXX", join_path(self.component_prefix.bin, "mpicxx")) env.set("MPIF77", join_path(self.component_prefix.bin, "mpif77")) env.set("MPIF90", join_path(self.component_prefix.bin, "mpif90")) env.set("MPIFC", join_path(self.component_prefix.bin, "mpifc")) else: env.set("MPICC", join_path(self.component_prefix.bin, "mpiicc")) env.set("MPICXX", join_path(self.component_prefix.bin, "mpiicpc")) env.set("MPIF77", join_path(self.component_prefix.bin, "mpiifort")) env.set("MPIF90", join_path(self.component_prefix.bin, "mpiifort")) env.set("MPIFC", join_path(self.component_prefix.bin, "mpiifort")) env.set("I_MPI_ROOT", self.component_prefix) @property def headers(self): return self.header_directories( [self.component_prefix.include, self.component_prefix.include.ilp64] ) @property def libs(self): libs = [] if "+ilp64" in self.spec: libs += find_libraries("libmpi_ilp64", self.component_prefix.lib.release) libs += find_libraries(["libmpicxx", "libmpifort"], self.component_prefix.lib) libs += find_libraries("libmpi", self.component_prefix.lib.release) libs += find_system_libraries(["libdl", "librt", "libpthread"]) # Find libfabric for libmpi.so if "+external-libfabric" in self.spec: libs += self.spec["libfabric"].libs else: libs += find_libraries(["libfabric"], self.component_prefix.libfabric.lib) return libs @run_after("install") def fix_wrappers(self): # When spack builds from source # fix I_MPI_SUBSTITUTE_INSTALLDIR and # __EXEC_PREFIX_TO_BE_FILLED_AT_INSTALL_TIME__ for wrapper in ["mpif77", "mpif90", "mpigcc", "mpigxx", "mpiicc", "mpiicpc", "mpiifort"]: filter_file( r"I_MPI_SUBSTITUTE_INSTALLDIR|" r"__EXEC_PREFIX_TO_BE_FILLED_AT_INSTALL_TIME__", self.component_prefix, self.component_prefix.bin.join(wrapper), backup=False, ) @run_after("install") def fixup_prefix(self): # The motivation was to provide a more standard layout so impi # would be more likely to work as a virtual dependence. It # does not work for v2_layout because of a library conflict. I # am not sure if this mechanism is useful so disabling for # v2_layout rather than try to make it work. if self.v2_layout: return self.symlink_dir(self.component_prefix.include, self.prefix.include) self.symlink_dir(self.component_prefix.lib, self.prefix.lib) self.symlink_dir(self.component_prefix.lib.release, self.prefix.lib) self.symlink_dir(self.component_prefix.bin, self.prefix.bin)