libclc/configure.py

   1 #!/usr/bin/python
   2
   3 def c_compiler_rule(b, name, description, compiler, flags):
   4   command = "%s -MMD -MF $out.d %s -c -o $out $in" % (compiler, flags)
   5   b.rule(name, command, description + " $out", depfile="$out.d")
   6
   7 from optparse import OptionParser
   8 import os
   9 from subprocess import *
  10 import sys
  11
  12 srcdir = os.path.dirname(sys.argv[0])
  13
  14 sys.path.insert(0, os.path.join(srcdir, 'build'))
  15 import metabuild
  16
  17 p = OptionParser()
  18 p.add_option('--with-llvm-config', metavar='PATH',
  19              help='use given llvm-config script')
  20 p.add_option('--prefix', metavar='PATH',
  21              help='install to given prefix')
  22 p.add_option('-g', metavar='GENERATOR', default='make',
  23              help='use given generator (default: make)')
  24 (options, args) = p.parse_args()
  25
  26 llvm_config_exe = options.with_llvm_config or "llvm-config"
  27
  28 def llvm_config(args):
  29   try:
  30     proc = Popen([llvm_config_exe] + args, stdout=PIPE)
  31     return proc.communicate()[0].rstrip().replace('\n', ' ')
  32   except OSError:
  33     print "Error executing llvm-config."
  34     print "Please ensure that llvm-config is in your $PATH, or use --with-llvm-config."
  35     sys.exit(1)
  36
  37 llvm_bindir = llvm_config(['--bindir'])
  38 llvm_core_libs = llvm_config(['--libs', 'core', 'bitreader', 'bitwriter']) + ' ' + \
  39                  llvm_config(['--ldflags'])
  40 llvm_cxxflags = llvm_config(['--cxxflags']) + ' -fno-exceptions -fno-rtti'
  41
  42 llvm_clang = os.path.join(llvm_bindir, 'clang')
  43 llvm_link = os.path.join(llvm_bindir, 'llvm-link')
  44 llvm_opt = os.path.join(llvm_bindir, 'opt')
  45
  46 default_targets = ['nvptx--nvidiacl', 'nvptx64--nvidiacl']
  47
  48 targets = args
  49 if not targets:
  50   targets = default_targets
  51
  52 b = metabuild.from_name(options.g)
  53
  54 b.rule("LLVM_AS", "%s -o $out $in" % os.path.join(llvm_bindir, "llvm-as"),
  55        'LLVM-AS $out')
  56 b.rule("LLVM_LINK", command = llvm_link + " -o $out $in",
  57        description = 'LLVM-LINK $out')
  58 b.rule("OPT", command = llvm_opt + " -O3 -o $out $in",
  59        description = 'OPT $out')
  60
  61 c_compiler_rule(b, "LLVM_TOOL_CXX", 'CXX', 'c++', llvm_cxxflags)
  62 b.rule("LLVM_TOOL_LINK", "c++ -o $out $in %s" % llvm_core_libs, 'LINK $out')
  63
  64 prepare_builtins = os.path.join('utils', 'prepare-builtins')
  65 b.build(os.path.join('utils', 'prepare-builtins.o'), "LLVM_TOOL_CXX",
  66         os.path.join(srcdir, 'utils', 'prepare-builtins.cpp'))
  67 b.build(prepare_builtins, "LLVM_TOOL_LINK",
  68         os.path.join('utils', 'prepare-builtins.o'))
  69
  70 b.rule("PREPARE_BUILTINS", "%s -o $out $in" % prepare_builtins,
  71        'PREPARE-BUILTINS $out')
  72
  73 manifest_deps = set([sys.argv[0], os.path.join(srcdir, 'build', 'metabuild.py'),
  74                      os.path.join(srcdir, 'build', 'ninja_syntax.py')])
  75
  76 install_files = []
  77 install_deps = []
  78
  79 for target in targets:
  80   (t_arch, t_vendor, t_os) = target.split('-')
  81   archs = [t_arch]
  82   if t_arch == 'nvptx' or t_arch == 'nvptx64':
  83     archs.append('ptx')
  84   archs.append('generic')
  85
  86   subdirs = []
  87   for arch in archs:
  88     subdirs.append("%s-%s-%s" % (arch, t_vendor, t_os))
  89     subdirs.append("%s-%s" % (arch, t_os))
  90     subdirs.append(arch)
  91
  92   incdirs = filter(os.path.isdir,
  93                [os.path.join(srcdir, subdir, 'include') for subdir in subdirs])
  94   libdirs = filter(lambda d: os.path.isfile(os.path.join(d, 'SOURCES')),
  95                    [os.path.join(srcdir, subdir, 'lib') for subdir in subdirs])
  96
  97   clang_cl_includes = ' '.join(["-I%s" % incdir for incdir in incdirs])
  98   install_files += [(incdir, incdir[len(srcdir)+1:]) for incdir in incdirs]
  99
 100   # The rule for building a .bc file for the specified architecture using clang.
 101   clang_bc_flags = "-target %s -I`dirname $in` %s " \
 102                    "-Dcl_clang_storage_class_specifiers " \
 103                    "-Dcl_khr_fp64 " \
 104                    "-emit-llvm" % (target, clang_cl_includes)
 105   clang_bc_rule = "CLANG_CL_BC_" + target
 106   c_compiler_rule(b, clang_bc_rule, "LLVM-CC", llvm_clang, clang_bc_flags)
 107
 108   objects = []
 109   sources_seen = set()
 110
 111   for libdir in libdirs:
 112     subdir_list_file = os.path.join(libdir, 'SOURCES')
 113     manifest_deps.add(subdir_list_file)
 114     for src in open(subdir_list_file).readlines():
 115       src = src.rstrip()
 116       if src not in sources_seen:
 117         sources_seen.add(src)
 118         obj = os.path.join(target, 'lib', src + '.bc')
 119         objects.append(obj)
 120         src_file = os.path.join(libdir, src)
 121         ext = os.path.splitext(src)[1]
 122         if ext == '.ll':
 123           b.build(obj, 'LLVM_AS', src_file)
 124         else:
 125           b.build(obj, clang_bc_rule, src_file)
 126
 127   builtins_link_bc = os.path.join(target, 'lib', 'builtins.link.bc')
 128   builtins_opt_bc = os.path.join(target, 'lib', 'builtins.opt.bc')
 129   builtins_bc = os.path.join(target, 'lib', 'builtins.bc')
 130   b.build(builtins_link_bc, "LLVM_LINK", objects)
 131   b.build(builtins_opt_bc, "OPT", builtins_link_bc)
 132   b.build(builtins_bc, "PREPARE_BUILTINS", builtins_opt_bc, prepare_builtins)
 133   install_files.append((builtins_bc, builtins_bc))
 134   install_deps.append(builtins_bc)
 135   b.default(builtins_bc)
 136
 137 if options.prefix:
 138   install_cmd = ' && '.join(['mkdir -p %(dst)s && cp -r %(src)s %(dst)s' %
 139                              {'src': file,
 140                               'dst': os.path.join(options.prefix,
 141                                                   os.path.dirname(dest))}
 142                              for (file, dest) in install_files])
 143   b.rule('install', command = install_cmd, description = 'INSTALL')
 144   b.build('install', 'install', install_deps)
 145
 146 b.rule("configure", command = ' '.join(sys.argv), description = 'CONFIGURE',
 147        generator = True)
 148 b.build(b.output_filename(), 'configure', list(manifest_deps))
 149
 150 b.finish()