1# -*- Python -*-
2
3import os
4
5# Setup config name.
6config.name = 'Profile'
7
8# Setup source root.
9config.test_source_root = os.path.dirname(__file__)
10
11# Setup executable root.
12if hasattr(config, 'profile_lit_binary_dir') and \
13        config.profile_lit_binary_dir is not None:
14    config.test_exec_root = config.profile_lit_binary_dir
15
16# If the above check didn't work, we're probably in the source tree.  Use some
17# magic to re-execute from the build tree.
18if config.test_exec_root is None:
19    # The magic relies on knowing compilerrt_site_basedir.
20    compilerrt_basedir = lit_config.params.get('compilerrt_site_basedir', None)
21    if compilerrt_basedir:
22        site_cfg = os.path.join(compilerrt_basedir, 'profile', 'lit.site.cfg')
23        if os.path.exists(site_cfg):
24            lit_config.load_config(config, site_cfg)
25            raise SystemExit
26
27# Test suffixes.
28config.suffixes = ['.c', '.cc', '.cpp', '.m', '.mm', '.ll', '.test']
29
30# What to exclude.
31config.excludes = ['Inputs']
32
33# Clang flags.
34clang_cflags = [config.target_cflags]
35
36def build_invocation(compile_flags):
37  return " " + " ".join([config.clang] + compile_flags) + " "
38
39# Add clang substitutions.
40config.substitutions.append( ("%clang ", build_invocation(clang_cflags)) )
41config.substitutions.append( ("%clang_profgen ", build_invocation(clang_cflags) + " -fprofile-instr-generate ") )
42config.substitutions.append( ("%clang_profuse=", build_invocation(clang_cflags) + " -fprofile-instr-use=") )
43
44# Profile tests are currently supported on Linux and Darwin only.
45if config.host_os not in ['Linux', 'Darwin']:
46  config.unsupported = True
47