aboutsummaryrefslogtreecommitdiffhomepage
path: root/third_party/sycl
diff options
context:
space:
mode:
authorGravatar Yifei Feng <yifeif@google.com>2017-11-22 00:33:29 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-11-22 00:36:36 -0800
commitad7eeec1cc06d7fdba6ee404f03a35fab9cd3e6a (patch)
tree455989721eda10ba8e4313a48ad0e44aa1f124db /third_party/sycl
parent0c98a7ecf88da45469cf00edc5cc4c0c82c7d49f (diff)
Automated g4 rollback of changelist 176615737
PiperOrigin-RevId: 176621645
Diffstat (limited to 'third_party/sycl')
-rw-r--r--third_party/sycl/crosstool/trisycl.tpl60
-rw-r--r--third_party/sycl/sycl_configure.bzl1
2 files changed, 25 insertions, 36 deletions
diff --git a/third_party/sycl/crosstool/trisycl.tpl b/third_party/sycl/crosstool/trisycl.tpl
index 87a70d8f95..b470772fbf 100644
--- a/third_party/sycl/crosstool/trisycl.tpl
+++ b/third_party/sycl/crosstool/trisycl.tpl
@@ -11,12 +11,10 @@ CPU_C_COMPILER = ('%{host_c_compiler}')
CURRENT_DIR = os.path.dirname(sys.argv[0])
TRISYCL_INCLUDE_DIR = CURRENT_DIR + '/../sycl/include'
-
def main():
compiler_flags = []
- remove_flags = ('-Wl,--no-undefined', '-Wno-unused-but-set-variable',
- '-Wignored-attributes', '-fno-exceptions')
+ remove_flags = ('-Wl,--no-undefined', '-Wno-unused-but-set-variable', '-Wignored-attributes', '-fno-exceptions')
# remove -fsamotoze-coverage from string with g++
if 'g++' in CPU_CXX_COMPILER:
remove_flags += ('-fsanitize-coverage',)
@@ -24,62 +22,52 @@ def main():
else:
compiler_flags += ['-fopenmp=libomp']
- compiler_flags += [
- flag for flag in sys.argv[1:] if not flag.startswith(remove_flags)
- ]
+ compiler_flags += [flag for flag in sys.argv[1:] if not flag.startswith(remove_flags)]
+
output_file_index = compiler_flags.index('-o') + 1
output_file_name = compiler_flags[output_file_index]
- if (output_file_index == 1):
+ if(output_file_index == 1):
# we are linking
- return call([CPU_CXX_COMPILER] + compiler_flags + ['-Wl,--no-undefined'])
+ return call([CPU_CXX_COMPILER] + compiler_flags +
+ ['-Wl,--no-undefined'])
# find what we compile
compiling_cpp = 0
- if ('-c' in compiler_flags):
- compiled_file_index = compiler_flags.index('-c') + 1
- compiled_file_name = compiler_flags[compiled_file_index]
- if (compiled_file_name.endswith(('.cc', '.c++', '.cpp', '.CPP', '.C',
- '.cxx'))):
- compiling_cpp = 1
-
- debug_flags = [
- '-DTRISYCL_DEBUG', '-DBOOST_LOG_DYN_LINK', '-DTRISYCL_TRACE_KERNEL',
- '-lpthread', '-lboost_log', '-g', '-rdynamic'
- ]
+ if('-c' in compiler_flags):
+ compiled_file_index = compiler_flags.index('-c') + 1
+ compiled_file_name = compiler_flags[compiled_file_index]
+ if(compiled_file_name.endswith(('.cc', '.c++', '.cpp', '.CPP',
+ '.C', '.cxx'))):
+ compiling_cpp = 1;
+
+ debug_flags = ['-DTRISYCL_DEBUG', '-DBOOST_LOG_DYN_LINK', '-DTRISYCL_TRACE_KERNEL', '-lpthread', '-lboost_log', '-g', '-rdynamic']
opt_flags = ['-DNDEBUG', '-DBOOST_DISABLE_ASSERTS', '-O3']
- compiler_flags = compiler_flags + [
- '-DEIGEN_USE_SYCL=1', '-DEIGEN_HAS_C99_MATH',
- '-DEIGEN_MAX_ALIGN_BYTES=16', '-DTENSORFLOW_USE_SYCL'
- ] + opt_flags
+ compiler_flags = compiler_flags + ['-DEIGEN_USE_SYCL=1',
+ '-DEIGEN_HAS_C99_MATH',
+ '-DEIGEN_MAX_ALIGN_BYTES=16',
+ '-DTENSORFLOW_USE_SYCL'] + opt_flags
- if (compiling_cpp == 1):
+ if(compiling_cpp == 1):
# create a blacklist of folders that will be skipped when compiling
# with triSYCL
- skip_extensions = ['.cu.cc']
- skip_folders = [
- 'tensorflow/compiler', 'tensorflow/docs_src', 'tensorflow/tensorboard',
- 'third_party', 'external', 'hexagon'
- ]
+ skip_extensions = [".cu.cc"]
+ skip_folders = ["tensorflow/compiler", "tensorflow/docs_src", "tensorflow/tensorboard", "third_party", "external", "hexagon"]
skip_folders = [(folder + '/') for folder in skip_folders]
# if compiling external project skip triSYCL
- if any(
- compiled_file_name.endswith(_ext) for _ext in skip_extensions) or any(
- _folder in output_file_name for _folder in skip_folders):
+ if any(compiled_file_name.endswith(_ext) for _ext in skip_extensions) or any(_folder in output_file_name for _folder in skip_folders):
return call([CPU_CXX_COMPILER] + compiler_flags)
- host_compiler_flags = [
- '-xc++', '-Wno-unused-variable', '-I', TRISYCL_INCLUDE_DIR
- ] + compiler_flags
+ host_compiler_flags = ['-xc++', '-Wno-unused-variable',
+ '-I', TRISYCL_INCLUDE_DIR] + compiler_flags
x = call([CPU_CXX_COMPILER] + host_compiler_flags)
return x
else:
# compile for C
return 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 5b9d0eb383..a0c9e4e43a 100644
--- a/third_party/sycl/sycl_configure.bzl
+++ b/third_party/sycl/sycl_configure.bzl
@@ -67,6 +67,7 @@ def find_computecpp_root(repository_ctx):
def find_trisycl_include_dir(repository_ctx):
"""Find triSYCL include directory. """
+ sycl_name = ""
if _TRISYCL_INCLUDE_DIR in repository_ctx.os.environ:
sycl_name = repository_ctx.os.environ[_TRISYCL_INCLUDE_DIR].strip()
if sycl_name.startswith("/"):