1licenses(["restricted"])
2
3load(":generate.bzl", "tensorflow_rbe_config")
4
5tensorflow_rbe_config(
6    name = "ubuntu16.04-py3-clang",
7    compiler = "clang",
8    os = "ubuntu16.04",
9    python_version = "3",
10)
11
12tensorflow_rbe_config(
13    name = "centos6-py-gcc7",
14    build_bazel_src = True,
15    compiler = "gcc",
16    compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
17    os = "centos6",
18    python_version = "2",
19)
20
21tensorflow_rbe_config(
22    name = "centos6-py3-gcc7",
23    build_bazel_src = True,
24    compiler = "gcc",
25    compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
26    os = "centos6",
27    python_version = "3",
28)
29
30tensorflow_rbe_config(
31    name = "centos6-py3-gcc7-cuda10.0-cudnn7-tensorrt5",
32    build_bazel_src = True,
33    compiler = "gcc",
34    compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
35    cuda_version = "10.0",
36    cudnn_install_path = "/usr/local/cuda-10.0",
37    cudnn_version = "7",
38    os = "centos6",
39    python_version = "3",
40    tensorrt_install_path = "/usr",
41    tensorrt_version = "5",
42)
43
44tensorflow_rbe_config(
45    name = "centos6-py3-gcc7-cuda10.1-cudnn7-tensorrt5",
46    build_bazel_src = True,
47    compiler = "gcc",
48    compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
49    cuda_version = "10.1",
50    cudnn_install_path = "/usr/local/cuda-10.1",
51    cudnn_version = "7",
52    os = "centos6",
53    python_version = "3",
54    tensorrt_install_path = "/usr",
55    tensorrt_version = "5",
56)
57
58tensorflow_rbe_config(
59    name = "ubuntu14.04-py3-gcc7-cuda10.0-cudnn7-tensorrt5",
60    compiler = "gcc-7",
61    cuda_version = "10.0",
62    cudnn_version = "7",
63    os = "ubuntu14.04",
64    python_version = "3",
65    tensorrt_version = "5",
66)
67
68tensorflow_rbe_config(
69    name = "ubuntu14.04-py3-gcc-cuda10.0-cudnn7-tensorrt5",
70    compiler = "gcc",
71    cuda_version = "10.0",
72    cudnn_version = "7",
73    os = "ubuntu14.04",
74    python_version = "3",
75    tensorrt_version = "5",
76)
77
78tensorflow_rbe_config(
79    name = "ubuntu16.04-py-gcc7_manylinux2010",
80    compiler = "/dt7/usr/bin/gcc",
81    compiler_prefix = "/usr/bin",
82    os = "ubuntu16.04-manylinux2010",
83    python_version = "2",
84)
85
86tensorflow_rbe_config(
87    name = "ubuntu16.04-py3-gcc7_manylinux2010",
88    compiler = "/dt7/usr/bin/gcc",
89    compiler_prefix = "/usr/bin",
90    os = "ubuntu16.04-manylinux2010",
91    python_version = "3.6",
92)
93
94tensorflow_rbe_config(
95    name = "ubuntu16.04-py3-clang_manylinux2010-cuda10.0-cudnn7-tensorrt5.1",
96    compiler = "/clang_r373795/bin/clang",
97    cuda_version = "10.0",
98    cudnn_version = "7",
99    os = "ubuntu16.04-manylinux2010",
100    python_version = "3",
101    sysroot = "/dt7",
102    tensorrt_install_path = "/usr",
103    tensorrt_version = "5.1",
104)
105
106tensorflow_rbe_config(
107    name = "ubuntu16.04-py3-gcc7_manylinux2010-cuda10.1-cudnn7-tensorrt6.0",
108    compiler = "/dt7/usr/bin/gcc",
109    compiler_prefix = "/usr/bin",
110    cuda_version = "10.1",
111    cudnn_version = "7",
112    os = "ubuntu16.04-manylinux2010",
113    python_version = "3",
114    tensorrt_install_path = "/usr",
115    tensorrt_version = "6.0",
116)
117
118tensorflow_rbe_config(
119    name = "ubuntu16.04-py3-clang_manylinux2010-cuda10.1-cudnn7-tensorrt6.0",
120    compiler = "/clang_r373795/bin/clang",
121    cuda_version = "10.1",
122    cudnn_version = "7",
123    os = "ubuntu16.04-manylinux2010",
124    python_version = "3",
125    sysroot = "/dt7",
126    tensorrt_install_path = "/usr",
127    tensorrt_version = "6.0",
128)
129