diff options
author | Martin Wicke <wicke@google.com> | 2016-12-14 15:46:53 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-12-14 16:04:11 -0800 |
commit | 2e4869af1afe55135d522142be3a2a483162a1b1 (patch) | |
tree | c676b228682e796ff0c2896b20faf97b66b8a1f2 /third_party/sycl | |
parent | 811629aed466db32eeefbd60783e199d2fe154a9 (diff) |
Merge changes from github.
Change: 142074581
Diffstat (limited to 'third_party/sycl')
-rwxr-xr-x | third_party/sycl/crosstool/computecpp.tpl | 42 | ||||
-rw-r--r-- | third_party/sycl/sycl_configure.bzl | 11 |
2 files changed, 31 insertions, 22 deletions
diff --git a/third_party/sycl/crosstool/computecpp.tpl b/third_party/sycl/crosstool/computecpp.tpl index 0c7611d298..e90d51bf87 100755 --- a/third_party/sycl/crosstool/computecpp.tpl +++ b/third_party/sycl/crosstool/computecpp.tpl @@ -13,22 +13,28 @@ COMPUTECPP_DRIVER= COMPUTECPP_ROOT + 'bin/compute++' COMPUTECPP_INCLUDE = COMPUTECPP_ROOT + 'include' def main(): - computecpp_compiler_flags = [flag for flag in sys.argv[1:]] + compiler_flags = [] - output_file_index = computecpp_compiler_flags.index('-o') + 1 - output_file_name = computecpp_compiler_flags[output_file_index] + # remove -fsamotoze-coverage from string + if CPU_CXX_COMPILER.find("g++") != -1: + compiler_flags = [flag for flag in sys.argv[1:] if not flag.startswith(('-Wl,--no-undefined', '-fsanitize-coverage', '-Wno-unused-but-set-variable', '-Wignored-attributes'))] + else: + compiler_flags = [flag for flag in sys.argv[1:] if not flag.startswith(('-Wl,--no-undefined', '-Wno-unused-but-set-variable', '-Wignored-attributes'))] + + output_file_index = compiler_flags.index('-o') + 1 + output_file_name = compiler_flags[output_file_index] if(output_file_index == 1): # we are linking - return subprocess.call([CPU_CXX_COMPILER] + computecpp_compiler_flags) + return subprocess.call([CPU_CXX_COMPILER] + compiler_flags) - computecpp_compiler_flags = computecpp_compiler_flags + ['-D_GLIBCXX_USE_CXX11_ABI=0'] + compiler_flags = compiler_flags + ['-D_GLIBCXX_USE_CXX11_ABI=0', '-DEIGEN_USE_SYCL=1'] # find what we compile compiling_cpp = 0 - if('-c' in computecpp_compiler_flags): - compiled_file_index = computecpp_compiler_flags.index('-c') + 1 - compited_file_name = computecpp_compiler_flags[compiled_file_index] + if('-c' in compiler_flags): + compiled_file_index = compiler_flags.index('-c') + 1 + compited_file_name = compiler_flags[compiled_file_index] if(compited_file_name.endswith(('.cc', '.c++', '.cpp', '.CPP', '.C', '.cxx'))): compiling_cpp = 1; @@ -36,22 +42,24 @@ def main(): filename, file_extension = os.path.splitext(output_file_name) bc_out = filename + '.sycl' - computecpp_compiler_flags = ['--sycl-no-diags', '-sycl-compress-name', '-DTENSORFLOW_USE_SYCL', '-Wno-unused-variable', '-I', COMPUTECPP_INCLUDE, '-isystem', - COMPUTECPP_INCLUDE, '-std=c++11', '-sycl', '-emit-llvm', '-no-serial-memop'] + computecpp_compiler_flags + # strip asan for the device + computecpp_device_compiler_flags = [flag for flag in compiler_flags if not flag.startswith(('-fsanitize'))] + computecpp_device_compiler_flags = ['-sycl-compress-name', '-DTENSORFLOW_USE_SYCL', '-Wno-unused-variable', '-I', COMPUTECPP_INCLUDE, '-isystem', + COMPUTECPP_INCLUDE, '-std=c++11', '-sycl', '-emit-llvm', '-no-serial-memop'] + computecpp_device_compiler_flags - # dont want that in case of compiling with computecpp first - host_compiler_flags = [flag for flag in sys.argv[1:] - if not flag.startswith(('-MF', '-MD',)) - if not '.d' in flag] - - x = subprocess.call([COMPUTECPP_DRIVER] + computecpp_compiler_flags ) + x = subprocess.call([COMPUTECPP_DRIVER] + computecpp_device_compiler_flags ) if(x == 0): + # dont want that in case of compiling with computecpp first + host_compiler_flags = [flag for flag in compiler_flags + if not flag.startswith(('-MF', '-MD',)) + if not '.d' in flag] + host_compiler_flags = ['-D_GLIBCXX_USE_CXX11_ABI=0', '-DTENSORFLOW_USE_SYCL', '-Wno-unused-variable', '-I', COMPUTECPP_INCLUDE, '--include', bc_out] + host_compiler_flags x = subprocess.call([CPU_CXX_COMPILER] + host_compiler_flags) return x else: # compile for C - return subprocess.call([CPU_C_COMPILER] + computecpp_compiler_flags) + return subprocess.call([CPU_C_COMPILER] + compiler_flags) if __name__ == '__main__': sys.exit(main()) diff --git a/third_party/sycl/sycl_configure.bzl b/third_party/sycl/sycl_configure.bzl index 38bd7759de..6ad498487f 100644 --- a/third_party/sycl/sycl_configure.bzl +++ b/third_party/sycl/sycl_configure.bzl @@ -102,7 +102,7 @@ def _tpl(repository_ctx, tpl, substitutions={}, out=None): def _file(repository_ctx, label): repository_ctx.template( label.replace(":", "/"), - Label("//third_party/sycl/%s.tpl" % label), + Label("//third_party/sycl/%s" % label), {}) _DUMMY_CROSSTOOL_BZL_FILE = """ @@ -133,9 +133,9 @@ error_sycl_disabled() def _create_dummy_repository(repository_ctx): # Set up BUILD file for sycl/. - _file(repository_ctx, "sycl:build_defs.bzl") + _tpl(repository_ctx, "sycl:build_defs.bzl") _tpl(repository_ctx, "sycl:BUILD") - _tpl(repository_ctx, "sycl:LICENSE.text") + _file(repository_ctx, "sycl:LICENSE.text") _tpl(repository_ctx, "sycl:platform.bzl") # Create dummy files for the SYCL toolkit since they are still required by @@ -157,10 +157,11 @@ def _sycl_autoconf_imp(repository_ctx): _create_dummy_repository(repository_ctx) else: # copy template files - _file(repository_ctx, "sycl:build_defs.bzl") + _tpl(repository_ctx, "sycl:build_defs.bzl") _tpl(repository_ctx, "sycl:BUILD") _tpl(repository_ctx, "sycl:platform.bzl") - _file(repository_ctx, "crosstool:BUILD") + _tpl(repository_ctx, "crosstool:BUILD") + _file(repository_ctx, "sycl:LICENSE.text") _tpl(repository_ctx, "crosstool:computecpp", { "%{host_cxx_compiler}" : find_cc(repository_ctx), |