6dd6ad9fd7
Currently, we set the tag "local" for GPU tests in order to be able to execute remote CPU and local GPU tests within the same bazel invocation. This change introduces the possibility to set REMOTE_GPU_TESTING to enable GPU tests to also run remotely; given that tags cannot use starlark's select, we use an autoconfig rule that defines a function returning the tags we want: "local" by default and "remote-gpu" if REMOTE_GPU_TESTING is set. The platform is set via exec_compatible_with constraints, so we select on the "remote-gpu" tag to add a constraint that is only fulfilled by GPU-enabled platforms. PiperOrigin-RevId: 229141861 |
||
|---|---|---|
| .. | ||
| BUILD | ||
| BUILD.tpl | ||
| configure.bzl | ||
| execution.bzl.tpl | ||