experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Internal change.

Browse Source 
PiperOrigin-RevId: 253321760
This commit is contained in:
Amit Patankar 2019-06-14 16:52:59 -07:00 committed by TensorFlower Gardener
parent 81655b9ad3
commit 58ed949448
13 changed files with 1686 additions and 109 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/opensource_only.files
View File

@ -65,6 +65,7 @@ tensorflow/third_party/toolchains/preconfig/centos6/py/BUILD
tensorflow/third_party/toolchains/preconfig/centos6/cuda10.0-cudnn7/cuda/BUILD
tensorflow/third_party/toolchains/preconfig/centos6/cuda10.0-cudnn7/cuda/build_defs.bzl
tensorflow/third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.0/BUILD
tensorflow/third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.0/cc_toolchain_config.bzl
tensorflow/third_party/toolchains/preconfig/centos6/tensorrt5/build_defs.bzl
tensorflow/third_party/toolchains/preconfig/centos6/tensorrt5/BUILD
tensorflow/third_party/toolchains/preconfig/centos6/gcc7/cc_toolchain_config.bzl

22
tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-centos6
View File

@ -16,8 +16,8 @@
# To work around this, we create a Docker environment in /tmp to build the
# image from in a shell script (2).
FROM nvidia/cuda:10.0-cudnn7-devel-centos6
LABEL maintainer="Manuel Klimek <klimek@google.com>"
FROM cudatest/test:10.0-cudnn7-devel-centos6-patched-glibc
LABEL maintainer="Amit Patankar <amitpatankar@google.com>"
# Install packages required to build tensorflow.
RUN yum install -y centos-release-scl && \
@ -37,26 +37,26 @@ RUN rpm -ihv tensorrt.rpm && \
yum install -y --nogpgcheck tensorrt && \
rm -f tensorrt.rpm
# Enable devtoolset-7 and rh-python35 in the docker image.
env PATH="/opt/rh/rh-python35/root/usr/bin:/opt/rh/devtoolset-7/root/usr/bin:/usr/local/nvidia/bin:/usr/local/cuda/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" \
LD_LIBRARY_PATH="/opt/rh/rh-python35/root/usr/lib64:/opt/rh/devtoolset-7/root/usr/lib64:/opt/rh/devtoolset-7/root/usr/lib:/opt/rh/devtoolset-7/root/usr/lib64/dyninst:/opt/rh/devtoolset-7/root/usr/lib/dyninst:/opt/rh/devtoolset-7/root/usr/lib64:/opt/rh/devtoolset-7/root/usr/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64" \
# Enable devtoolset-7 and python27 in the docker image.
env PATH="/opt/rh/python27/root/usr/bin:/opt/rh/devtoolset-7/root/usr/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin" \
LD_LIBRARY_PATH="/opt/rh/python27/root/usr/lib64:/opt/rh/devtoolset-7/root/usr/lib64:/opt/rh/devtoolset-7/root/usr/lib:/opt/rh/devtoolset-7/root/usr/lib64/dyninst:/opt/rh/devtoolset-7/root/usr/lib/dyninst:/opt/rh/devtoolset-7/root/usr/lib64:/opt/rh/devtoolset-7/root/usr/lib:/usr/local/nvidia/lib:/usr/local/nvidia/lib64" \
PCP_DIR="/opt/rh/devtoolset-7/root" \
PERL5LIB="/opt/rh/devtoolset-7/root//usr/lib64/perl5/vendor_perl:/opt/rh/devtoolset-7/root/usr/lib/perl5:/opt/rh/devtoolset-7/root//usr/share/perl5/vendor_perl" \
PKG_CONFIG_PATH="/opt/rh/rh-python35/root/usr/lib64/pkgconfig/"
PKG_CONFIG_PATH="/opt/rh/python27/root/usr/lib64/pkgconfig/"
# Install pip packages needed to build tensorflow.
COPY install/*.sh /install/
RUN grep -E 'easy_install |pip3' install/install_pip_packages.sh \
| grep -v 'no-binary=:all:' \
> install/centos_pip_packages.sh && \
bash install/centos_pip_packages.sh
RUN bash install/install_yum_packages.sh
RUN bash install/install_centos_python36.sh
RUN bash install/install_centos_pip_packages.sh
# Install a /usr/bin/python3 link.
# centos by default does not provide links, and instead relies on paths into
# /opt/ to switch to alternative configurations. For bazel remote builds,
# the python path between the local machine running bazel and the remote setup
# must be the same.
RUN update-alternatives --install /usr/bin/python3 python3 /opt/rh/rh-python35/root/usr/bin/python3 0
RUN update-alternatives --install /usr/bin/python2 python2 /opt/rh/python27/root/usr/bin/python2.7 0
RUN update-alternatives --install /usr/bin/python3 python3 /usr/local/bin/python3.6 0
# Install a ubuntu-compatible openjdk link so that ubuntu JAVA_HOME works
# for this image.

2
tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-centos6.sh
View File

@ -34,7 +34,7 @@
WORKDIR="$(mktemp -d)"
BASE="$(pwd)"
cp -R "${BASE}/"* "${WORKDIR}/"
cp "/tmp/nv-tensorrt-repo-rhel7-cuda10.0-trt5.0.2.6-ga-20181010-1-1.x86_64.rpm" "${WORKDIR}/tensorrt.rpm"
cp "/tmp/nv-tensorrt-repo-rhel7-cuda10.0-trt5.1.5.0-ga-20190427-1-1.x86_64.rpm" "${WORKDIR}/tensorrt.rpm"
cd "${WORKDIR}"
docker build -f Dockerfile.rbe.cuda10.0-cudnn7-centos6 \
--tag "gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-centos6" .

6
tensorflow/tools/ci_build/install/install_yum_packages.sh
View File

@ -43,4 +43,8 @@ yum install -y atlas-devel \
sqlite-devel \
wget \
xz-devel \
zlib-devel
zlib-devel
# Install latest git.
yum install -y http://opensource.wandisco.com/centos/6/git/x86_64/wandisco-git-release-6-1.noarch.rpm
yum install -y git

128
third_party/toolchains/preconfig/centos6/cuda10.0-cudnn7/cuda/BUILD vendored
View File

@ -1,3 +1,5 @@
load(":build_defs.bzl", "cuda_header_library")
licenses(["restricted"]) # MPL2, portions GPL v3, LGPL v3, BSD-like
package(default_visibility = ["//visibility:public"])
@ -28,37 +30,29 @@ config_setting(
config_setting(
name = "darwin",
values = {"cpu": "darwin"},
visibility = ["//visibility:public"],
)
config_setting(
name = "freebsd",
values = {"cpu": "freebsd"},
visibility = ["//visibility:public"],
)
cc_library(
cuda_header_library(
name = "cuda_headers",
hdrs = [
"cuda/cuda_config.h",
":cuda-include",
":cudnn-include",
],
include_prefix = "third_party/gpus",
includes = [
".",
".", # required to include cuda/cuda/cuda_config.h as cuda/config.h
"cuda/include",
"cuda/include/crt",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "cudart_static",
srcs = ["cuda/lib/libcudart_static.a"],
includes = [
".",
"cuda/include",
],
linkopts = select({
":freebsd": [],
"//conditions:default": ["-ldl"],
@ -66,104 +60,75 @@ cc_library(
"-lpthread",
"-lrt",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "cuda_driver",
srcs = ["cuda/lib/libcuda.so"],
includes = [
".",
"cuda/include",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "cudart",
srcs = ["cuda/lib/libcudart.so.10.0"],
data = ["cuda/lib/libcudart.so.10.0"],
includes = [
".",
"cuda/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cuda_header_library(
name = "cublas_headers",
hdrs = [":cublas-include"],
include_prefix = "third_party/gpus/cuda/include",
includes = ["cublas/include"],
strip_include_prefix = "cublas/include",
deps = [":cuda_headers"],
)
cc_library(
name = "cublas",
srcs = ["cuda/lib/libcublas.so.10.0"],
data = ["cuda/lib/libcublas.so.10.0"],
includes = [
".",
"cuda/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "cusolver",
srcs = ["cuda/lib/libcusolver.so.10.0"],
data = ["cuda/lib/libcusolver.so.10.0"],
includes = [
".",
"cuda/include",
],
linkopts = ["-lgomp"],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "cudnn",
srcs = ["cuda/lib/libcudnn.so.7"],
data = ["cuda/lib/libcudnn.so.7"],
includes = [
".",
"cuda/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "cudnn_header",
includes = [
".",
"cuda/include",
],
visibility = ["//visibility:public"],
hdrs = [":cudnn-include"],
include_prefix = "third_party/gpus/cudnn",
strip_include_prefix = "cudnn/include",
deps = [":cuda_headers"],
)
cc_library(
name = "cufft",
srcs = ["cuda/lib/libcufft.so.10.0"],
data = ["cuda/lib/libcufft.so.10.0"],
includes = [
".",
"cuda/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "curand",
srcs = ["cuda/lib/libcurand.so.10.0"],
data = ["cuda/lib/libcurand.so.10.0"],
includes = [
".",
"cuda/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "cuda",
visibility = ["//visibility:public"],
deps = [
":cublas",
":cuda_headers",
@ -174,46 +139,30 @@ cc_library(
],
)
cc_library(
cuda_header_library(
name = "cupti_headers",
hdrs = [
"cuda/cuda_config.h",
":cuda-extras",
],
includes = [
".",
"cuda/extras/CUPTI/include/",
],
visibility = ["//visibility:public"],
hdrs = [":cuda-extras"],
include_prefix = "third_party/gpus",
includes = ["cuda/extras/CUPTI/include/"],
deps = [":cuda_headers"],
)
cc_library(
name = "cupti_dsos",
data = ["cuda/lib/libcupti.so.10.0"],
includes = [
".",
"cuda/include",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "cusparse",
srcs = ["cuda/lib/libcusparse.so.10.0"],
data = ["cuda/lib/libcusparse.so.10.0"],
includes = [
".",
"cuda/include",
],
linkopts = ["-lgomp"],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "libdevice_root",
data = [":cuda-nvvm"],
visibility = ["//visibility:public"],
)
genrule(
@ -1254,6 +1203,18 @@ genrule(
cmd = """cp -rLf "/usr/local/cuda-10.0/extras/CUPTI/include/." "$(@D)/cuda/extras/CUPTI/include/" """,
)
genrule(
name = "cublas-include",
outs = [
"cublas/include/cublas.h",
"cublas/include/cublas_v2.h",
"cublas/include/cublas_api.h",
],
cmd = """cp -f "/usr/local/cuda-10.0/include/cublas.h" "$(location cublas/include/cublas.h)" && \
cp -f "/usr/local/cuda-10.0/include/cublas_v2.h" "$(location cublas/include/cublas_v2.h)" && \
cp -f "/usr/local/cuda-10.0/include/cublas_api.h" "$(location cublas/include/cublas_api.h)" """,
)
genrule(
name = "cuda-lib",
outs = [
@ -1268,7 +1229,16 @@ genrule(
"cuda/lib/libcupti.so.10.0",
"cuda/lib/libcusparse.so.10.0",
],
cmd = """cp -f "/usr/local/cuda-10.0/lib64/stubs/libcuda.so" $(location cuda/lib/libcuda.so) && cp -f "/usr/local/cuda-10.0/lib64/libcudart.so.10.0" $(location cuda/lib/libcudart.so.10.0) && cp -f "/usr/local/cuda-10.0/lib64/libcudart_static.a" $(location cuda/lib/libcudart_static.a) && cp -f "/usr/local/cuda-10.0/lib64/libcublas.so.10.0" $(location cuda/lib/libcublas.so.10.0) && cp -f "/usr/local/cuda-10.0/lib64/libcusolver.so.10.0" $(location cuda/lib/libcusolver.so.10.0) && cp -f "/usr/local/cuda-10.0/lib64/libcurand.so.10.0" $(location cuda/lib/libcurand.so.10.0) && cp -f "/usr/local/cuda-10.0/lib64/libcufft.so.10.0" $(location cuda/lib/libcufft.so.10.0) && cp -f "/usr/local/cuda-10.0/lib64/libcudnn.so.7" $(location cuda/lib/libcudnn.so.7) && cp -f "/usr/local/cuda-10.0/extras/CUPTI/lib64/libcupti.so.10.0" $(location cuda/lib/libcupti.so.10.0) && cp -f "/usr/local/cuda-10.0/lib64/libcusparse.so.10.0" $(location cuda/lib/libcusparse.so.10.0) """,
cmd = """cp -f "/usr/local/cuda-10.0/lib64/stubs/libcuda.so" "$(location cuda/lib/libcuda.so)" && \
cp -f "/usr/local/cuda-10.0/lib64/libcudart.so.10.0" "$(location cuda/lib/libcudart.so.10.0)" && \
cp -f "/usr/local/cuda-10.0/lib64/libcudart_static.a" "$(location cuda/lib/libcudart_static.a)" && \
cp -f "/usr/local/cuda-10.0/lib64/libcublas.so.10.0" "$(location cuda/lib/libcublas.so.10.0)" && \
cp -f "/usr/local/cuda-10.0/lib64/libcusolver.so.10.0" "$(location cuda/lib/libcusolver.so.10.0)" && \
cp -f "/usr/local/cuda-10.0/lib64/libcurand.so.10.0" "$(location cuda/lib/libcurand.so.10.0)" && \
cp -f "/usr/local/cuda-10.0/lib64/libcufft.so.10.0" "$(location cuda/lib/libcufft.so.10.0)" && \
cp -f "/usr/local/cuda-10.0/lib64/libcudnn.so.7" "$(location cuda/lib/libcudnn.so.7)" && \
cp -f "/usr/local/cuda-10.0/extras/CUPTI/lib64/libcupti.so.10.0" "$(location cuda/lib/libcupti.so.10.0)" && \
cp -f "/usr/local/cuda-10.0/lib64/libcusparse.so.10.0" "$(location cuda/lib/libcusparse.so.10.0)" """,
)
genrule(
@ -1295,4 +1265,10 @@ genrule(
cmd = """cp -rLf "/usr/local/cuda-10.0/bin/." "$(@D)/cuda/bin/" """,
)
filegroup(name = "cudnn-include")
genrule(
name = "cudnn-include",
outs = [
"cudnn/include/cudnn.h",
],
cmd = """cp -f "/usr/local/cuda-10.0/include/cudnn.h" "$(location cudnn/include/cudnn.h)" """,
)

33
third_party/toolchains/preconfig/centos6/cuda10.0-cudnn7/cuda/build_defs.bzl vendored
View File

@ -27,5 +27,34 @@ def if_cuda_is_configured(x):
--config=cuda. Used to allow non-CUDA code to depend on CUDA libraries.
"""
if cuda_is_configured():
return x
return []
return select({"//conditions:default": x})
return select({"//conditions:default": []})
def cuda_header_library(
name,
hdrs,
include_prefix = None,
strip_include_prefix = None,
deps = [],
**kwargs):
"""Generates a cc_library containing both virtual and system include paths.
Generates both a header-only target with virtual includes plus the full
target without virtual includes. This works around the fact that bazel can't
mix 'includes' and 'include_prefix' in the same target."""
native.cc_library(
name = name + "_virtual",
hdrs = hdrs,
include_prefix = include_prefix,
strip_include_prefix = strip_include_prefix,
deps = deps,
visibility = ["//visibility:private"],
)
native.cc_library(
name = name,
textual_hdrs = hdrs,
deps = deps + [":%s_virtual" % name],
**kwargs
)

0
third_party/toolchains/preconfig/centos6/cuda10.0-cudnn7/cuda/cuda/cuda_config.h vendored Normal file → Executable file
View File

86
third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.0/BUILD vendored
View File

@ -1,3 +1,8 @@
# This file is expanded from a template by cuda_configure.bzl
# Update cuda_configure.bzl#verify_build_defines when adding new variables.
load(":cc_toolchain_config.bzl", "cc_toolchain_config")
licenses(["restricted"])
package(default_visibility = ["//visibility:public"])
@ -24,6 +29,7 @@ cc_toolchain_suite(
"x64_windows|msvc-cl": ":cc-compiler-windows",
"x64_windows": ":cc-compiler-windows",
"arm": ":cc-compiler-local",
"aarch64": ":cc-compiler-local",
"k8": ":cc-compiler-local",
"piii": ":cc-compiler-local",
"ppc": ":cc-compiler-local",
@ -35,7 +41,6 @@ cc_toolchain(
name = "cc-compiler-local",
all_files = ":crosstool_wrapper_driver_is_not_gcc",
compiler_files = ":empty",
cpu = "local",
dwp_files = ":empty",
linker_files = ":crosstool_wrapper_driver_is_not_gcc",
objcopy_files = ":empty",
@ -45,35 +50,110 @@ cc_toolchain(
# last on the command line and contain all shared libraries to link, so all
# regular options will be left of them.
supports_param_files = 1,
toolchain_config = ":cc-compiler-local-config",
toolchain_identifier = "local_linux",
)
cc_toolchain_config(
name = "cc-compiler-local-config",
builtin_include_directories = [
"/opt/rh/devtoolset-7/root/usr/include/c++/7",
"/opt/rh/devtoolset-7/root/usr/include/c++/7/x86_64-redhat-linux",
"/opt/rh/devtoolset-7/root/usr/include/c++/7/backward",
"/opt/rh/devtoolset-7/root/usr/lib/gcc/x86_64-redhat-linux/7/include",
"/usr/local/include",
"/opt/rh/devtoolset-7/root/usr/include",
"/usr/include",
"/usr/local/cuda-10.0/targets/x86_64-linux/include",
"/usr/local/cuda-10.0/include",
"/usr/local/cuda-10.0/extras/CUPTI/include",
"/usr/local/cuda-10.0/include",
],
cpu = "local",
extra_no_canonical_prefixes_flags = ["-fno-canonical-system-headers"],
host_compiler_path = "clang/bin/crosstool_wrapper_driver_is_not_gcc",
host_compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
host_compiler_warnings = [],
host_unfiltered_compile_flags = [],
linker_bin_path = "/opt/rh/devtoolset-7/root/usr/bin",
)
cc_toolchain(
name = "cc-compiler-darwin",
all_files = ":crosstool_wrapper_driver_is_not_gcc",
compiler_files = ":empty",
cpu = "darwin",
dwp_files = ":empty",
linker_files = ":crosstool_wrapper_driver_is_not_gcc",
objcopy_files = ":empty",
strip_files = ":empty",
supports_param_files = 0,
toolchain_config = ":cc-compiler-local-darwin",
toolchain_identifier = "local_darwin",
)
cc_toolchain_config(
name = "cc-compiler-local-darwin",
builtin_include_directories = [
"/opt/rh/devtoolset-7/root/usr/include/c++/7",
"/opt/rh/devtoolset-7/root/usr/include/c++/7/x86_64-redhat-linux",
"/opt/rh/devtoolset-7/root/usr/include/c++/7/backward",
"/opt/rh/devtoolset-7/root/usr/lib/gcc/x86_64-redhat-linux/7/include",
"/usr/local/include",
"/opt/rh/devtoolset-7/root/usr/include",
"/usr/include",
"/usr/local/cuda-10.0/targets/x86_64-linux/include",
"/usr/local/cuda-10.0/include",
"/usr/local/cuda-10.0/extras/CUPTI/include",
"/usr/local/cuda-10.0/include",
],
cpu = "darwin",
extra_no_canonical_prefixes_flags = ["-fno-canonical-system-headers"],
host_compiler_path = "clang/bin/crosstool_wrapper_driver_is_not_gcc",
host_compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
host_compiler_warnings = [],
host_unfiltered_compile_flags = [],
linker_bin_path = "/opt/rh/devtoolset-7/root/usr/bin",
)
cc_toolchain(
name = "cc-compiler-windows",
all_files = ":windows_msvc_wrapper_files",
compiler_files = ":empty",
cpu = "x64_windows",
dwp_files = ":empty",
linker_files = ":windows_msvc_wrapper_files",
objcopy_files = ":empty",
strip_files = ":empty",
supports_param_files = 1,
toolchain_config = ":cc-compiler-windows-config",
toolchain_identifier = "local_windows",
)
cc_toolchain_config(
name = "cc-compiler-windows-config",
builtin_include_directories = [
"/opt/rh/devtoolset-7/root/usr/include/c++/7",
"/opt/rh/devtoolset-7/root/usr/include/c++/7/x86_64-redhat-linux",
"/opt/rh/devtoolset-7/root/usr/include/c++/7/backward",
"/opt/rh/devtoolset-7/root/usr/lib/gcc/x86_64-redhat-linux/7/include",
"/usr/local/include",
"/opt/rh/devtoolset-7/root/usr/include",
"/usr/include",
"/usr/local/cuda-10.0/targets/x86_64-linux/include",
"/usr/local/cuda-10.0/include",
"/usr/local/cuda-10.0/extras/CUPTI/include",
"/usr/local/cuda-10.0/include",
],
cpu = "x64_windows",
msvc_cl_path = "msvc_not_used",
msvc_env_include = "msvc_not_used",
msvc_env_lib = "msvc_not_used",
msvc_env_path = "msvc_not_used",
msvc_env_tmp = "msvc_not_used",
msvc_lib_path = "msvc_not_used",
msvc_link_path = "msvc_not_used",
msvc_ml_path = "msvc_not_used",
)
filegroup(
name = "empty",
srcs = [],

1486
third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.0/cc_toolchain_config.bzl vendored Executable file
View File

File diff suppressed because it is too large Load Diff

5
third_party/toolchains/preconfig/centos6/gcc7-nvcc-cuda10.0/clang/bin/crosstool_wrapper_driver_is_not_gcc vendored
View File

@ -1,4 +1,4 @@
#!/usr/bin/env python
#!/usr/bin/env python2
# Copyright 2015 The TensorFlow Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
@ -95,6 +95,7 @@ def GetHostCompilerOptions(argv):
parser.add_argument('--sysroot', nargs=1)
parser.add_argument('-g', nargs='*', action='append')
parser.add_argument('-fno-canonical-system-headers', action='store_true')
parser.add_argument('-no-canonical-prefixes', action='store_true')
args, _ = parser.parse_known_args(argv)
@ -108,6 +109,8 @@ def GetHostCompilerOptions(argv):
opts += ' -g' + ' -g'.join(sum(args.g, []))
if args.fno_canonical_system_headers:
opts += ' -fno-canonical-system-headers'
if args.no_canonical_prefixes:
opts += ' -no-canonical-prefixes'
if args.sysroot:
opts += ' --sysroot ' + args.sysroot[0]

21
third_party/toolchains/preconfig/centos6/tensorrt5/BUILD vendored
View File

@ -12,22 +12,16 @@ exports_files(["LICENSE"])
cc_library(
name = "tensorrt_headers",
hdrs = [":tensorrt_include"],
visibility = ["//visibility:public"],
include_prefix = "third_party/tensorrt",
strip_include_prefix = "tensorrt/include",
)
cc_library(
name = "tensorrt",
srcs = [
"tensorrt/lib/libnvinfer.so.5",
"tensorrt/lib/libnvinfer_plugin.so.5",
],
srcs = [":tensorrt_lib"],
copts = cuda_default_copts(),
data = [
"tensorrt/lib/libnvinfer.so.5",
"tensorrt/lib/libnvinfer_plugin.so.5",
],
data = [":tensorrt_lib"],
linkstatic = 1,
visibility = ["//visibility:public"],
deps = [
":tensorrt_headers",
"@local_config_cuda//cuda",
@ -40,7 +34,8 @@ genrule(
"tensorrt/lib/libnvinfer.so.5",
"tensorrt/lib/libnvinfer_plugin.so.5",
],
cmd = """cp -f "/usr/lib64/libnvinfer.so.5" $(location tensorrt/lib/libnvinfer.so.5) && cp -f "/usr/lib64/libnvinfer_plugin.so.5" $(location tensorrt/lib/libnvinfer_plugin.so.5) """,
cmd = """cp -f "/usr/lib64/libnvinfer.so.5" "$(location tensorrt/lib/libnvinfer.so.5)" && \
cp -f "/usr/lib64/libnvinfer_plugin.so.5" "$(location tensorrt/lib/libnvinfer_plugin.so.5)" """,
)
genrule(
@ -50,5 +45,7 @@ genrule(
"tensorrt/include/NvUtils.h",
"tensorrt/include/NvInferPlugin.h",
],
cmd = """cp -f "/usr/include/NvInfer.h" $(location tensorrt/include/NvInfer.h) && cp -f "/usr/include/NvUtils.h" $(location tensorrt/include/NvUtils.h) && cp -f "/usr/include/NvInferPlugin.h" $(location tensorrt/include/NvInferPlugin.h) """,
cmd = """cp -f "/usr/include/NvInfer.h" "$(location tensorrt/include/NvInfer.h)" && \
cp -f "/usr/include/NvUtils.h" "$(location tensorrt/include/NvUtils.h)" && \
cp -f "/usr/include/NvInferPlugin.h" "$(location tensorrt/include/NvInferPlugin.h)" """,
)

3
third_party/toolchains/preconfig/generate/BUILD vendored
View File

@ -29,6 +29,7 @@ tensorflow_rbe_config(
tensorflow_rbe_config(
name = "centos6-py3-gcc7-cuda10.0-cudnn7-tensorrt5",
build_bazel_src = True,
compiler = "gcc",
compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
cuda_version = "10.0",
@ -36,7 +37,7 @@ tensorflow_rbe_config(
cudnn_version = "7",
os = "centos6",
python_version = "3",
tensorrt_install_path = "/usr/lib64",
tensorrt_install_path = "/usr",
tensorrt_version = "5",
)

2
third_party/toolchains/preconfig/generate/containers.bzl vendored
View File

@ -4,6 +4,6 @@ container_digests = {
"centos6": "sha256:5450c88850260a5d46f4942c3807e2240eb5e792fce5e788bacf2317309969f1",
"cuda10.0-cudnn7-ubuntu14.04": "sha256:d433e1221f802dac393bc8652fabcc63aa46896cd920bb888ae0e2002fe6b756",
"cuda10.0-cudnn7-centos7": "sha256:a453b7147a60928a8345689eae48916a746b3578b5e831bfa151f0529d469c88",
"cuda10.0-cudnn7-centos6": "sha256:a874e7a102abb7da5861dd468b68a25f360bb976a39a26c1e123b770e7900322",
"cuda10.0-cudnn7-centos6": "sha256:671757b382068522e2a94acacb35281c435e80964217173af103ee1b594fd4df",
"cuda10.0-cudnn7-ubuntu16.04-manylinux2010": "sha256:eedcedfe63a778068bf725f9ffa425646725faac9ba96a57abfad307e832dcf9",
}