github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/toolchains/BUILD (about)

     1  licenses(["restricted"])
     2  
     3  package(default_visibility = ["//visibility:public"])
     4  
     5  load("//third_party/toolchains/preconfig/generate:containers.bzl", "container_digests")
     6  
     7  # Constraint used for platforms below so we can force certain rules to be executed
     8  # on specific platforms.
     9  constraint_setting(name = "custom_platforms")
    10  
    11  # Constraint for platforms that allow GPU testing (i.e. have a GPU available).
    12  # This is used in exec_compatible_with of rules that need GPU access.
    13  constraint_value(
    14      name = "gpu_test",
    15      constraint_setting = ":custom_platforms",
    16  )
    17  
    18  # Remote build platforms.
    19  # Each of the platform rules here provide a platform definition that is bound to a docker image.
    20  # The result of the skylark configuration is checked into
    21  # //tensorflow/third_party/toolchains/preconfig.
    22  
    23  # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cpu.
    24  platform(
    25      name = "rbe_ubuntu16.04",
    26      constraint_values = [
    27          "@bazel_tools//platforms:x86_64",
    28          "@bazel_tools//platforms:linux",
    29      ],
    30      remote_execution_properties = """
    31          properties: {
    32              name: "container-image"
    33              value:"docker://gcr.io/tensorflow-testing/nosla-ubuntu16.04@%s"
    34          }
    35          properties: {
    36              name: "Pool"
    37              value: "default"
    38          }
    39          """ % container_digests["ubuntu16.04"],
    40  )
    41  
    42  # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cpu-centos6.
    43  platform(
    44      name = "rbe_centos6",
    45      constraint_values = [
    46          "@bazel_tools//platforms:x86_64",
    47          "@bazel_tools//platforms:linux",
    48      ],
    49      remote_execution_properties = """
    50          properties: {
    51              name: "container-image"
    52              value:"docker://gcr.io/tensorflow-testing/nosla-centos6@%s"
    53          }
    54          properties: {
    55              name: "Pool"
    56              value: "default"
    57          }
    58          """ % container_digests["centos6"],
    59  )
    60  
    61  # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-ubuntu14.04.
    62  platform(
    63      name = "rbe_cuda10.0-cudnn7-ubuntu14.04",
    64      constraint_values = [
    65          "@bazel_tools//platforms:x86_64",
    66          "@bazel_tools//platforms:linux",
    67      ],
    68      remote_execution_properties = """
    69          properties: {
    70              name: "container-image"
    71              value:"docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu14.04@%s"
    72          }
    73          properties: {
    74              name: "Pool"
    75              value: "default"
    76          }
    77          """ % container_digests["cuda10.0-cudnn7-ubuntu14.04"],
    78  )
    79  
    80  # The above platform with GPU support.
    81  platform(
    82      name = "rbe_cuda10.0-cudnn7-ubuntu14.04-gpu",
    83      constraint_values = [
    84          "@bazel_tools//platforms:x86_64",
    85          "@bazel_tools//platforms:linux",
    86          ":gpu_test",
    87      ],
    88      remote_execution_properties = """
    89          properties: {
    90              name: "container-image"
    91              value: "docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu14.04@%s"
    92          }
    93          properties: {
    94              name: "dockerRuntime"
    95              value: "nvidia"
    96          }
    97          properties: {
    98              name: "Pool"
    99              value: "gpu-pool"
   100          }
   101          """ % container_digests["cuda10.0-cudnn7-ubuntu14.04"],
   102  )
   103  
   104  # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-centos6.
   105  platform(
   106      name = "rbe_cuda10.0-cudnn7-centos6",
   107      constraint_values = [
   108          "@bazel_tools//platforms:x86_64",
   109          "@bazel_tools//platforms:linux",
   110      ],
   111      remote_execution_properties = """
   112          properties: {
   113              name: "container-image"
   114              value:"docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-centos6@%s"
   115          }
   116          properties: {
   117              name: "Pool"
   118              value: "default"
   119          }
   120          """ % container_digests["cuda10.0-cudnn7-centos6"],
   121  )
   122  
   123  # The above platform with GPU support.
   124  platform(
   125      name = "rbe_cuda10.0-cudnn7-centos6-gpu",
   126      constraint_values = [
   127          "@bazel_tools//platforms:x86_64",
   128          "@bazel_tools//platforms:linux",
   129          ":gpu_test",
   130      ],
   131      remote_execution_properties = """
   132          properties: {
   133              name: "container-image"
   134              value: "docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-centos6@%s"
   135          }
   136          properties: {
   137              name: "dockerRuntime"
   138              value: "nvidia"
   139          }
   140          properties: {
   141              name: "Pool"
   142              value: "gpu-pool"
   143          }
   144          """ % container_digests["cuda10.0-cudnn7-centos6"],
   145  )
   146  
   147  # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.ubuntu16.04-manylinux2010.
   148  platform(
   149      name = "rbe_ubuntu16.04-manylinux2010",
   150      constraint_values = [
   151          "@bazel_tools//platforms:x86_64",
   152          "@bazel_tools//platforms:linux",
   153      ],
   154      remote_execution_properties = """
   155          properties: {
   156              name: "container-image"
   157              value:"docker://gcr.io/tensorflow-testing/nosla-ubuntu16.04-manylinux2010@%s"
   158          }
   159          properties: {
   160              name: "Pool"
   161              value: "default"
   162          }
   163          """ % container_digests["ubuntu16.04-manylinux2010"],
   164  )
   165  
   166  # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-ubuntu16.04-manylinux2010.
   167  platform(
   168      name = "rbe_cuda10.0-cudnn7-ubuntu16.04-manylinux2010",
   169      constraint_values = [
   170          "@bazel_tools//platforms:x86_64",
   171          "@bazel_tools//platforms:linux",
   172      ],
   173      remote_execution_properties = """
   174          properties: {
   175              name: "container-image"
   176              value:"docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010@%s"
   177          }
   178          properties: {
   179              name: "Pool"
   180              value: "default"
   181          }
   182          """ % container_digests["cuda10.0-cudnn7-ubuntu16.04-manylinux2010"],
   183  )
   184  
   185  # The above platform with GPU support.
   186  platform(
   187      name = "rbe_cuda10.0-cudnn7-ubuntu16.04-manylinux2010-gpu",
   188      constraint_values = [
   189          "@bazel_tools//platforms:x86_64",
   190          "@bazel_tools//platforms:linux",
   191          ":gpu_test",
   192      ],
   193      remote_execution_properties = """
   194          properties: {
   195              name: "container-image"
   196              value: "docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010@%s"
   197          }
   198          properties: {
   199              name: "dockerRuntime"
   200              value: "nvidia"
   201          }
   202          properties: {
   203              name: "Pool"
   204              value: "gpu-pool"
   205          }
   206          """ % container_digests["cuda10.0-cudnn7-ubuntu16.04-manylinux2010"],
   207  )
   208  
   209  # Built with //tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu16.04
   210  platform(
   211      name = "rbe_rocm-ubuntu16.04",
   212      constraint_values = [
   213          "@bazel_tools//platforms:x86_64",
   214          "@bazel_tools//platforms:linux",
   215      ],
   216      remote_execution_properties = """
   217          properties: {
   218              name: "container-image"
   219              value:"docker://gcr.io/tensorflow-testing/nosla-rocm-ubuntu16.04@%s"
   220          }
   221          properties: {
   222              name: "Pool"
   223              value: "default"
   224          }
   225          """ % container_digests["rocm-ubuntu16.04"],
   226  )