llvm/test/Bindings/Go/lit.local.cfg

   1 import os
   2 import pipes
   3 import shlex
   4 import sys
   5
   6 if not 'go' in config.root.llvm_bindings:
   7     config.unsupported = True
   8
   9 if not config.root.include_go_tests:
  10     config.unsupported = True
  11
  12 if config.have_tf_aot or config.have_tf_api:
  13     config.unsupported = True
  14
  15 def find_executable(executable, path=None):
  16     if path is None:
  17         path = os.environ['PATH']
  18     paths = path.split(os.pathsep)
  19     base, ext = os.path.splitext(executable)
  20
  21     if (sys.platform == 'win32' or os.name == 'os2') and (ext != '.exe'):
  22         executable = executable + '.exe'
  23
  24     if not os.path.isfile(executable):
  25         for p in paths:
  26             f = os.path.join(p, executable)
  27             if os.path.isfile(f):
  28                 return f
  29         return None
  30     else:
  31         return executable
  32
  33 # Resolve certain symlinks in the first word of compiler.
  34 #
  35 # This is a Go-specific hack. cgo and other Go tools check $CC and $CXX for the
  36 # substring 'clang' to determine if the compiler is Clang. This won't work if
  37 # $CC is cc and cc is a symlink pointing to clang, as it is on Darwin.
  38 #
  39 # Go tools also have problems with ccache, so we disable it.
  40 def fixup_compiler_path(compiler):
  41     args = shlex.split(compiler)
  42     if args[0].endswith('ccache') or args[0].endswith('gomacc'):
  43         args = args[1:]
  44
  45     path = find_executable(args[0])
  46
  47     try:
  48         if path.endswith('/cc') and os.readlink(path) == 'clang':
  49             args[0] = path[:len(path)-2] + 'clang'
  50     except (AttributeError, OSError):
  51         pass
  52
  53     try:
  54         if path.endswith('/c++') and os.readlink(path) == 'clang++':
  55             args[0] = path[:len(path)-3] + 'clang++'
  56     except (AttributeError, OSError):
  57         pass
  58
  59     return ' '.join([pipes.quote(arg) for arg in args])
  60
  61 config.environment['CC'] = fixup_compiler_path(config.host_cc)
  62 config.environment['CXX'] = fixup_compiler_path(config.host_cxx)
  63 config.environment['CGO_LDFLAGS'] = config.host_ldflags