licenses(["restricted"])

load(":generate.bzl", "tensorflow_rbe_config")

tensorflow_rbe_config(
    name = "ubuntu14.04-py3-gcc-cuda9.0-cudnn7-nccl2",
    compiler = "gcc",
    cuda_version = "9.0",
    cudnn_version = "7",
    python_version = "3",
)

tensorflow_rbe_config(
    name = "ubuntu14.04-py3-clang-cuda9.0-cudnn7-nccl2",
    compiler = "clang",
    cuda_version = "9.0",
    cudnn_version = "7",
    python_version = "3",
)

tensorflow_rbe_config(
    name = "ubuntu14.04-py3-gcc-cuda10.0-cudnn7-nccl2",
    compiler = "gcc",
    cuda_version = "10.0",
    cudnn_version = "7",
    python_version = "3",
)

tensorflow_rbe_config(
    name = "ubuntu14.04-py3-clang-cuda10.0-cudnn7-nccl2",
    compiler = "clang",
    cuda_version = "10.0",
    cudnn_version = "7",
    python_version = "3",
)
