forked from OSchip/llvm-project
133 lines
5.2 KiB
INI
133 lines
5.2 KiB
INI
# -*clang- Python -*-
|
|
|
|
import os
|
|
import platform
|
|
import re
|
|
import subprocess
|
|
|
|
import lit.formats
|
|
import lit.util
|
|
|
|
# Configuration file for the 'lit' test runner.
|
|
|
|
# name: The name of this test suite.
|
|
config.name = 'Polly'
|
|
|
|
# testFormat: The test format to use to interpret tests.
|
|
#
|
|
# For now we require '&&' between commands, until they get globally killed and
|
|
# the test runner updated.
|
|
execute_external = platform.system() != 'Windows'
|
|
config.test_format = lit.formats.ShTest(execute_external)
|
|
|
|
# suffixes: A list of file extensions to treat as test files.
|
|
config.suffixes = ['.ll']
|
|
|
|
# test_source_root: The root path where tests are located.
|
|
config.test_source_root = os.path.dirname(__file__)
|
|
|
|
# test_exec_root: The root path where tests should be run.
|
|
polly_obj_root = getattr(config, 'polly_obj_root', None)
|
|
if polly_obj_root is not None:
|
|
config.test_exec_root = os.path.join(polly_obj_root, 'test')
|
|
|
|
# Set llvm_{src,obj}_root for use by others.
|
|
config.llvm_src_root = getattr(config, 'llvm_src_root', None)
|
|
config.llvm_obj_root = getattr(config, 'llvm_obj_root', None)
|
|
|
|
# Tweak the PATH to include the tools dir and the scripts dir.
|
|
if polly_obj_root is not None:
|
|
llvm_tools_dir = getattr(config, 'llvm_tools_dir', None)
|
|
if not llvm_tools_dir:
|
|
lit_config.fatal('No LLVM tools dir set!')
|
|
extra_paths = getattr(config, 'extra_paths', [])
|
|
base_paths = [llvm_tools_dir, config.environment['PATH']]
|
|
path = os.path.pathsep.join(base_paths + extra_paths)
|
|
config.environment['PATH'] = path
|
|
|
|
llvm_libs_dir = getattr(config, 'llvm_libs_dir', None)
|
|
if not llvm_libs_dir:
|
|
lit_config.fatal('No LLVM libs dir set!')
|
|
path = os.path.pathsep.join((llvm_libs_dir,
|
|
config.environment.get('LD_LIBRARY_PATH','')))
|
|
config.environment['LD_LIBRARY_PATH'] = path
|
|
|
|
###
|
|
|
|
# Check that the object root is known.
|
|
if config.test_exec_root is None:
|
|
# Otherwise, we haven't loaded the site specific configuration (the user is
|
|
# probably trying to run on a test file directly, and either the site
|
|
# configuration hasn't been created by the build system, or we are in an
|
|
# out-of-tree build situation).
|
|
|
|
# Check for 'polly_site_config' user parameter, and use that if available.
|
|
site_cfg = lit_config.params.get('polly_site_config', None)
|
|
if site_cfg and os.path.exists(site_cfg):
|
|
lit_config.load_config(config, site_cfg)
|
|
raise SystemExit
|
|
|
|
# Try to detect the situation where we are using an out-of-tree build by
|
|
# looking for 'llvm-config'.
|
|
#
|
|
# FIXME: I debated (i.e., wrote and threw away) adding logic to
|
|
# automagically generate the lit.site.cfg if we are in some kind of fresh
|
|
# build situation. This means knowing how to invoke the build system though,
|
|
# and I decided it was too much magic. We should solve this by just having
|
|
# the .cfg files generated during the configuration step.
|
|
|
|
llvm_config = lit.util.which('llvm-config', config.environment['PATH'])
|
|
if not llvm_config:
|
|
lit_config.fatal('No site specific configuration available!')
|
|
|
|
# Get the source and object roots.
|
|
llvm_src_root = subprocess.check_output(['llvm-config', '--src-root']).decode("utf-8").strip()
|
|
llvm_obj_root = subprocess.check_output(['llvm-config', '--obj-root']).decode("utf-8").strip()
|
|
|
|
polly_src_root = os.path.join(llvm_src_root, "tools", "polly")
|
|
polly_obj_root = os.path.join(llvm_obj_root, "tools", "polly")
|
|
|
|
# Validate that we got a tree which points to here, using the standard
|
|
# tools/polly layout.
|
|
this_src_root = os.path.dirname(config.test_source_root)
|
|
if os.path.realpath(polly_src_root) != os.path.realpath(this_src_root):
|
|
lit_config.fatal('No site specific configuration available!')
|
|
|
|
# Check that the site specific configuration exists.
|
|
site_cfg = os.path.join(polly_obj_root, 'test', 'lit.site.cfg')
|
|
if not os.path.exists(site_cfg):
|
|
lit_config.fatal('No site specific configuration available!')
|
|
|
|
# Okay, that worked. Notify the user of the automagic, and reconfigure.
|
|
lit_config.note('using out-of-tree build at %r' % polly_obj_root)
|
|
lit_config.load_config(config, site_cfg)
|
|
raise SystemExit
|
|
|
|
# opt knows whether it is compiled with -DNDEBUG.
|
|
import subprocess
|
|
try:
|
|
opt_cmd = subprocess.Popen([os.path.join(llvm_tools_dir, 'opt'), '-version'],
|
|
stdout = subprocess.PIPE,
|
|
env=config.environment)
|
|
except OSError:
|
|
print("Could not find opt in " + llvm_tools_dir)
|
|
exit(42)
|
|
|
|
if re.search(r'with assertions', opt_cmd.stdout.read().decode('ascii')):
|
|
config.available_features.add('asserts')
|
|
opt_cmd.wait()
|
|
|
|
try:
|
|
llvm_config_cmd = subprocess.Popen([os.path.join(llvm_tools_dir,
|
|
'llvm-config'),
|
|
'--targets-built'],
|
|
stdout = subprocess.PIPE,
|
|
env=config.environment)
|
|
except OSError:
|
|
print("Could not find llvm-config in " + llvm_tools_dir)
|
|
exit(42)
|
|
|
|
if re.search(r'NVPTX', llvm_config_cmd.stdout.read().decode('ascii')):
|
|
config.available_features.add('nvptx-registered-target')
|
|
llvm_config_cmd.wait()
|