1f52a8e250
- remote_execution_properties has been deprecated in favor of exec_properties [1]. - have gpu platforms inherit from the cpu platform in order to avoid duplication. - Replace all uses of the exec_compatible_with attribute for GPU tests with the exec_properties attribute. exec_properties are passed through as platform properties to the remote execution backend. This makes creating special platforms for GPU tests obsolete. [1] https://docs.bazel.build/versions/master/be/platform.html#platform.exec_properties PiperOrigin-RevId: 292407784 Change-Id: I809354b37f4952ae73d3c4d9bfd9a9b7c14527bf
115 lines
4.0 KiB
Python
115 lines
4.0 KiB
Python
licenses(["restricted"])
|
|
|
|
package(default_visibility = ["//visibility:public"])
|
|
|
|
load("//third_party/toolchains/preconfig/generate:containers.bzl", "container_digests")
|
|
|
|
# Remote build platforms.
|
|
# Each of the platform rules here provide a platform definition that is bound to a docker image.
|
|
# The result of the skylark configuration is checked into
|
|
# //tensorflow/third_party/toolchains/preconfig.
|
|
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cpu.
|
|
platform(
|
|
name = "rbe_ubuntu16.04",
|
|
constraint_values = [
|
|
"@bazel_tools//platforms:x86_64",
|
|
"@bazel_tools//platforms:linux",
|
|
],
|
|
exec_properties = {
|
|
"container-image": "docker://gcr.io/tensorflow-testing/nosla-ubuntu16.04@%s" % container_digests["ubuntu16.04"],
|
|
"Pool": "default",
|
|
},
|
|
)
|
|
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cpu-centos6.
|
|
platform(
|
|
name = "rbe_centos6",
|
|
constraint_values = [
|
|
"@bazel_tools//platforms:x86_64",
|
|
"@bazel_tools//platforms:linux",
|
|
],
|
|
exec_properties = {
|
|
"container-image": "docker://gcr.io/tensorflow-testing/nosla-centos6@%s" % container_digests["centos6"],
|
|
"Pool": "default",
|
|
},
|
|
)
|
|
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-ubuntu14.04.
|
|
platform(
|
|
name = "rbe_cuda10.0-cudnn7-ubuntu14.04",
|
|
constraint_values = [
|
|
"@bazel_tools//platforms:x86_64",
|
|
"@bazel_tools//platforms:linux",
|
|
],
|
|
exec_properties = {
|
|
"container-image": "docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu14.04@%s" % container_digests["cuda10.0-cudnn7-ubuntu14.04"],
|
|
"Pool": "default",
|
|
},
|
|
)
|
|
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-centos6.
|
|
platform(
|
|
name = "rbe_cuda10.0-cudnn7-centos6",
|
|
constraint_values = [
|
|
"@bazel_tools//platforms:x86_64",
|
|
"@bazel_tools//platforms:linux",
|
|
],
|
|
exec_properties = {
|
|
"container-image": "docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-centos6@%s" % container_digests["cuda10.0-cudnn7-centos6"],
|
|
"Pool": "default",
|
|
},
|
|
)
|
|
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.ubuntu16.04-manylinux2010.
|
|
platform(
|
|
name = "rbe_ubuntu16.04-manylinux2010",
|
|
constraint_values = [
|
|
"@bazel_tools//platforms:x86_64",
|
|
"@bazel_tools//platforms:linux",
|
|
],
|
|
exec_properties = {
|
|
"container-image": "docker://gcr.io/tensorflow-testing/nosla-ubuntu16.04-manylinux2010@%s" % container_digests["ubuntu16.04-manylinux2010"],
|
|
"Pool": "default",
|
|
},
|
|
)
|
|
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.0-cudnn7-ubuntu16.04-manylinux2010.
|
|
platform(
|
|
name = "rbe_cuda10.0-cudnn7-ubuntu16.04-manylinux2010",
|
|
constraint_values = [
|
|
"@bazel_tools//platforms:x86_64",
|
|
"@bazel_tools//platforms:linux",
|
|
],
|
|
exec_properties = {
|
|
"container-image": "docker://gcr.io/tensorflow-testing/nosla-cuda10.0-cudnn7-ubuntu16.04-manylinux2010@%s" % container_digests["cuda10.0-cudnn7-ubuntu16.04-manylinux2010"],
|
|
"Pool": "default",
|
|
},
|
|
)
|
|
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.cuda10.1-cudnn7-ubuntu16.04-manylinux2010.
|
|
platform(
|
|
name = "rbe_cuda10.1-cudnn7-ubuntu16.04-manylinux2010",
|
|
constraint_values = [
|
|
"@bazel_tools//platforms:x86_64",
|
|
"@bazel_tools//platforms:linux",
|
|
],
|
|
exec_properties = {
|
|
"container-image": "docker://gcr.io/tensorflow-testing/nosla-cuda10.1-cudnn7-ubuntu16.04-manylinux2010@%s" % container_digests["cuda10.1-cudnn7-ubuntu16.04-manylinux2010"],
|
|
"Pool": "default",
|
|
},
|
|
)
|
|
|
|
# Built with //tensorflow/tools/ci_build/Dockerfile.rbe.rocm-ubuntu16.04
|
|
platform(
|
|
name = "rbe_rocm-ubuntu16.04",
|
|
constraint_values = [
|
|
"@bazel_tools//platforms:x86_64",
|
|
"@bazel_tools//platforms:linux",
|
|
],
|
|
exec_properties = {
|
|
"container-image": "docker://gcr.io/tensorflow-testing/nosla-rocm-ubuntu16.04@%s" % container_digests["rocm-ubuntu16.04"],
|
|
"Pool": "default",
|
|
},
|
|
)
|