def get_compile_args(libfabric=chpl_libfabric.get()):
flags = []
if libfabric == 'libfabric':
flags = third_party_utils.default_get_compile_args(
'libfabric', ucp=get_uniq_cfg_path())
elif libfabric == 'system':
# Allow overriding pkg-config via LIBFABRIC_DIR, for platforms
# without pkg-config.
libfab_dir_val = overrides.get('LIBFABRIC_DIR')
if libfab_dir_val:
flags.append('-I' + libfab_dir_val + '/include')
else:
# Try using pkg-config to get the compile-time flags.
pcflags = third_party_utils.pkgconfig_get_compile_args('libfabric',
system=True)
for pcl in pcflags:
flags.append(pcl)
launcher_val = chpl_launcher.get()
ofi_oob_val = overrides.get_environ('CHPL_RT_COMM_OFI_OOB')
if 'mpi' in launcher_val or (ofi_oob_val and 'mpi' in ofi_oob_val):
mpi_dir_val = overrides.get_environ('MPI_DIR')
if mpi_dir_val:
flags.append('-I' + mpi_dir_val + '/include')
return flags
def get_link_args(libfabric=chpl_libfabric.get()):
libs = []
if libfabric == 'libfabric':
return third_party_utils.default_get_link_args('libfabric',
ucp=get_uniq_cfg_path(),
libs=['libfabric.la'],
add_L_opt=True)
elif libfabric == 'system':
# Allow overriding pkg-config via LIBFABRIC_DIR, for platforms
# without pkg-config.
libfab_dir_val = overrides.get('LIBFABRIC_DIR')
if libfab_dir_val:
libs.extend([
'-L' + libfab_dir_val + '/lib',
'-Wl,-rpath,' + libfab_dir_val + '/lib', '-lfabric'
])
else:
# Try using pkg-config to get the libraries to link
# libfabric with.
pclibs = third_party_utils.pkgconfig_get_link_args('libfabric',
system=True)
for pcl in pclibs:
libs.append(pcl)
if pcl.startswith('-L'):
libs.append(pcl.replace('-L', '-Wl,-rpath,', 1))
return libs
def gather_pe_chpl_pkgconfig_libs():
# Don't do anything if we aren't using a PrgEnv compiler
if chpl_compiler.get_prgenv_compiler() == 'none':
return ""
import chpl_comm, chpl_comm_substrate, chpl_aux_filesys, chpl_libfabric
platform = chpl_platform.get('target')
comm = chpl_comm.get()
substrate = chpl_comm_substrate.get()
auxfs = chpl_aux_filesys.get()
ret = os.environ.get('PE_CHAPEL_PKGCONFIG_LIBS', '')
if comm != 'none':
if platform != 'hpe-cray-ex':
# Adding -lhugetlbfs gets the PrgEnv driver to add the appropriate
# linker option for static linking with it. While it's not always
# used with Chapel programs, it is expected to be the common case
# when running on a Cray X*, so just always linking it is ok.
# (We don't add it for HPE Cray EX systems, where it's not needed.)
ret = 'craype-hugetlbfs:cray-pmi:' + ret
if platform.startswith('cray-x'):
ret = 'cray-ugni:' + ret
if comm == 'gasnet' and substrate == 'aries':
ret = 'cray-udreg:' + ret
if comm == 'ofi' and chpl_libfabric.get() == 'system':
ret = 'libfabric:' + ret
# on login/compute nodes, lustre requires the devel api to make
# lustre/lustreapi.h available (it's implicitly available on esl nodes)
if 'lustre' in auxfs:
exists, returncode, out, err = try_run_command(
['pkg-config', '--exists', 'cray-lustre-api-devel'])
if exists and returncode == 0:
ret = 'cray-lustre-api-devel:' + ret
return ret