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