| # -*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() |