def get_compile_args():
args = ([], [])
libfabric_val = get()
if libfabric_val == 'bundled':
ucp_val = get_uniq_cfg_path()
args = third_party_utils.get_bundled_compile_args('libfabric',
ucp=ucp_val)
elif libfabric_val == 'system':
flags = []
# Allow overriding pkg-config via LIBFABRIC_DIR, for platforms
# without pkg-config.
libfab_dir_val = overrides.get('LIBFABRIC_DIR')
if libfab_dir_val:
args[1].append('-I' + libfab_dir_val + '/include')
else:
# Try using pkg-config to get the compile-time flags.
x = third_party_utils.pkgconfig_get_system_compile_args(
'libfabric')
args = x
if libfabric_val == 'system' or libfabric_val == 'bundled':
flags = []
launcher_val = chpl_launcher.get()
ofi_oob_val = overrides.get_environ('CHPL_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')
args[1].extend(flags)
return args
def get_compile_args(libfabric=get()):
flags = []
if libfabric == 'bundled':
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_compile_args(libfabric):
flags = []
if libfabric == 'system':
# 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)
elif libfabric == 'libfabric':
error("CHPL_LIBFABRIC=libfabric is not yet supported", ValueError)
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():
args = ([], [])
libfabric_val = get()
if libfabric_val == 'bundled':
args = third_party_utils.pkgconfig_get_bundled_link_args(
'libfabric', ucp=get_uniq_cfg_path())
elif libfabric_val == 'system':
libs = []
# 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.
tup = third_party_utils.pkgconfig_get_system_link_args('libfabric')
# put the two lists together (but expect tup[0] to be empty)
pclibs = tup[0] + tup[1]
# add -Wl,-rpath for the -L options
# this was a workaround and is probably not needed anymore
for pcl in pclibs:
libs.append(pcl)
if pcl.startswith('-L'):
libs.append(pcl.replace('-L', '-Wl,-rpath,', 1))
args[1].extend(libs)
if libfabric_val == 'system' or libfabric_val == 'bundled':
libs = []
launcher_val = chpl_launcher.get()
ofi_oob_val = overrides.get_environ('CHPL_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:
mpi_lib_dir = os.path.join(mpi_dir_val, 'lib64')
if not os.path.exists(mpi_lib_dir):
mpi_lib_dir = os.path.join(mpi_dir_val, 'lib')
if not os.path.exists(mpi_lib_dir):
mpi_lib_dir = None
if mpi_lib_dir:
libs.append('-L' + mpi_lib_dir)
libs.append('-Wl,-rpath,' + mpi_lib_dir)
mpi_lib_name = 'mpi'
if glob.glob(mpi_lib_dir + '/libmpich.*'):
mpi_lib_name = 'mpich'
libs.append('-l' + mpi_lib_name)
# If we're using the PMI2 out-of-band support we have to reference
# libpmi2 explicitly, except on Cray XC systems.
platform_val = chpl_platform.get('target')
if platform_val == 'hpe-cray-ex' or ofi_oob_val == 'pmi2':
libs.append('-lpmi2')
args[1].extend(libs)
return args
values=['fifo', 'qthreads'],
default=chpl_tasks.get(),
help_text='Tasks ({var_name}) values to build.',
),
Dimension(
'mem',
'CHPL_MEM',
values=['cstdlib', 'jemalloc'],
default=chpl_mem.get('target'),
help_text='Memory allocator ({var_name}) values to build.',
),
Dimension(
'launcher',
'CHPL_LAUNCHER',
values=['none', 'pbs-aprun', 'aprun', 'slurm-srun'],
default=chpl_launcher.get(),
help_text='Launcher ({var_name}) to build.',
),
Dimension(
'gmp',
'CHPL_GMP',
values=['none', 'gmp', 'system'],
default=chpl_gmp.get(),
help_text='GMP ({var_name}) values to build.',
),
Dimension(
'regexp',
'CHPL_REGEXP',
values=['none', 're2'],
default=chpl_regexp.get(),
help_text='Regular expression ({var_name}) values to build.',
Dimension(
'task', 'CHPL_TASKS',
values=['fifo', 'qthreads'],
default=chpl_tasks.get(),
help_text='Tasks ({var_name}) values to build.',
),
Dimension(
'mem', 'CHPL_MEM',
values=['cstdlib', 'tcmalloc', 'jemalloc', 'dlmalloc'],
default=chpl_mem.get('target'),
help_text='Memory allocator ({var_name}) values to build.',
),
Dimension(
'launcher', 'CHPL_LAUNCHER',
values=['none', 'pbs-aprun', 'aprun', 'slurm-srun'],
default=chpl_launcher.get(),
help_text='Launcher ({var_name}) to build.',
),
Dimension(
'gmp', 'CHPL_GMP',
values=['none', 'gmp', 'system'],
default=chpl_gmp.get(),
help_text='GMP ({var_name}) values to build.',
),
Dimension(
'regexp', 'CHPL_REGEXP',
values=['none', 're2'],
default=chpl_regexp.get(),
help_text='Regular expression ({var_name}) values to build.',
),
Dimension(
