1"""Configurations of RBE builds used with remote config."""
2
3load("//third_party/toolchains/remote_config:rbe_config.bzl", "tensorflow_local_config", "tensorflow_rbe_config", "tensorflow_rbe_win_config")
4
5def initialize_rbe_configs():
6    tensorflow_local_config(
7        name = "local_execution",
8    )
9
10    tensorflow_rbe_config(
11        name = "ubuntu16.04-manylinux2010-py3",
12        os = "ubuntu16.04-manylinux2010",
13        python_versions = ["3"],
14        compiler = "",
15    )
16
17    tensorflow_rbe_config(
18        name = "ubuntu16.04-py3-gcc7_manylinux2010-cuda10.0-cudnn7-tensorrt5.1",
19        compiler = "/dt7/usr/bin/gcc",
20        compiler_prefix = "/usr/bin",
21        cuda_version = "10.0",
22        cudnn_version = "7",
23        os = "ubuntu16.04-manylinux2010",
24        python_versions = ["3"],
25        tensorrt_install_path = "/usr",
26        tensorrt_version = "5.1",
27    )
28
29    tensorflow_rbe_config(
30        name = "ubuntu16.04-gcc7_manylinux2010-cuda10.1-cudnn7-tensorrt6.0",
31        compiler = "/dt7/usr/bin/gcc",
32        compiler_prefix = "/usr/bin",
33        cuda_version = "10.1",
34        cudnn_version = "7",
35        os = "ubuntu16.04-manylinux2010-multipython",
36        python_versions = ["2.7", "3.5", "3.6", "3.7", "3.8"],
37        tensorrt_install_path = "/usr",
38        tensorrt_version = "6.0",
39        python_install_path = "/usr/local",
40    )
41
42    tensorflow_rbe_config(
43        name = "ubuntu18.04-gcc7_manylinux2010-cuda10.1-cudnn7-tensorrt6.0",
44        compiler = "/dt7/usr/bin/gcc",
45        compiler_prefix = "/usr/bin",
46        cuda_version = "10.1",
47        cudnn_version = "7",
48        os = "ubuntu18.04-manylinux2010-multipython",
49        python_versions = ["2.7", "3.5", "3.6", "3.7", "3.8"],
50        tensorrt_install_path = "/usr",
51        tensorrt_version = "6.0",
52        python_install_path = "/usr/local",
53    )
54
55    tensorflow_rbe_config(
56        name = "ubuntu18.04-gcc7_manylinux2010-cuda11.0-cudnn8-tensorrt7.1",
57        compiler = "/dt7/usr/bin/gcc",
58        compiler_prefix = "/usr/bin",
59        cuda_version = "11.0",
60        cudnn_version = "8",
61        os = "ubuntu18.04-manylinux2010-multipython",
62        python_versions = ["2.7", "3.5", "3.6", "3.7", "3.8"],
63        tensorrt_install_path = "/usr",
64        tensorrt_version = "7.1",
65        python_install_path = "/usr/local",
66    )
67
68    # TODO(klimek): Delete this once all users are migrated to a python-version
69    # independent configuration. In the future, use
70    # "ubuntu16.04-gcc7_manylinux2010-cuda10.1-cudnn7-tensorrt6.0" instead.
71    tensorflow_rbe_config(
72        name = "ubuntu16.04-py3-gcc7_manylinux2010-cuda10.1-cudnn7-tensorrt6.0",
73        compiler = "/dt7/usr/bin/gcc",
74        compiler_prefix = "/usr/bin",
75        cuda_version = "10.1",
76        cudnn_version = "7",
77        os = "ubuntu16.04-manylinux2010",
78        python_versions = ["3"],
79        tensorrt_install_path = "/usr",
80        tensorrt_version = "6.0",
81    )
82
83    tensorflow_rbe_config(
84        name = "ubuntu18.04-clang_manylinux2010-cuda11.0-cudnn8-tensorrt7.1",
85        compiler = "/clang_r7f6f9f4cf966c78a315d15d6e913c43cfa45c47c/bin/clang",
86        cuda_version = "11.0",
87        cudnn_version = "8",
88        os = "ubuntu18.04-manylinux2010-multipython",
89        python_versions = ["2.7", "3.5", "3.6", "3.7", "3.8"],
90        tensorrt_install_path = "/usr",
91        tensorrt_version = "7.1",
92        sysroot = "/dt7",
93        python_install_path = "/usr/local",
94    )
95
96    tensorflow_rbe_config(
97        name = "ubuntu18.04-gcc7_manylinux2010-rocm",
98        compiler = "/dt7/usr/bin/gcc",
99        compiler_prefix = "/usr/bin",
100        rocm_version = "3.5",  # Any version will do.
101        os = "ubuntu18.04-manylinux2010-multipython",
102        python_versions = ["2.7", "3.5", "3.6", "3.7", "3.8"],
103        python_install_path = "/usr/local",
104    )
105
106    tensorflow_rbe_win_config(
107        name = "windows_py37",
108        python_bin_path = "C:/Python37/python.exe",
109    )
110